[SWEA D2] 1979 어디에 단어가 들어갈 수 있을까 (Python)

※ SW Expert 아카데미의 문제를 무단 복제하는 것을 금지합니다.

문제

SW Expert Academy

N X N 크기의 단어 퍼즐의 모양이 주어졌을 때 특정 길이 K를 갖는 단어가 들어갈 수 있는 자리의 수를 출력하는 프로그램

N = 5, K = 3 일 때, 길이가 3 인 단어가 들어갈 수 있는 자리는 2 곳(가로 1번, 가로 4번)이 된다.

퍼즐의 각 셀 중, 흰색 부분은 1, 검은색 부분은 0 으로 주어진다.

풀이

T = int(input())
for test_case in range(1, T + 1):
    n, k = map(int, input().split())
    puzzle = [list(map(int, input().split())) for _ in range(n)]
    answer = 0
    for i in range(n):
        count = 0
        # 행검사
        for j in range(n):
            if puzzle[i][j] == 1:
                count += 1
            if puzzle[i][j] == 0 or j == n-1:
                if count == k:
                    answer += 1
                count = 0
                
        # 열검사
        for j in range(n):
            if puzzle[j][i] == 1:
                count += 1
            if puzzle[j][i] == 0 or j == n-1:
                if count == k:
                    answer += 1
                count = 0                
    print(f"#{test_case} {answer}")