[SW Expert Academy] 1979. 어디에 단어가 들어갈 수 있을까

solution.py

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)]
    result = 0
    
   # 행을 기준으로 cnt
    for i in range(N):
        row = 0
        for j in range(N):
            if puzzle[i][j]==0:
                if row == K:
                    result += 1
                row = 0
            else:
                row += 1
        if row == K:
            result +=1
            
	# 열을 기준으로 cnt
    for j in range(N):
        col = 0
        for i in range(N):
            if puzzle[i][j]==0:
                if col == K:
                    result+= 1
                col = 0
            else:
                col += 1
        if col == K:
            result += 1
    print("#{} {}".format(test_case, result))