[SWEA] 1979 어디에 단어가 들어갈 수 있을까

문제의 저작권은 SWEA에 있습니다.
swea 1979 어디에 단어가 들어갈 수 있을까

문제를 너무 깊게 고민했다가 안풀린 문제.
처음에는 뭐 바깥으로 나가는 엣지 고민하고 단어 갯수와 단어갯수보다 큰 빈공간이 있는 것 처리하다가 꼬여서 싹 지우고 다시 생각했다.
그냥 행우선, 열우선으로 각각 이중 포문을 싹 돌면서, 연속된 1의 갯수를 세고 그 갯수가 단어의 길이와 맞는지만 체크하면 되는 문제였다.

import sys

sys.stdin = open("input (2).txt")

T = int(input())

for tc in range(1, T + 1):
    N, K = map(int, input().split())
    matrix = []
    for _ in range(N):
        matrix.append(list(map(int, input().split())))
    total = 0
    
    # 행우선
    for col in range(N):
        count = 0
        for row in range(N):
            if matrix[col][row] == 0:
                if count == K:
                    total += 1
                count = 0
            else:
                count += 1
        if count == K:
            total += 1
            
    # 열우선
    for row in range(N):
        count = 0
        for col in range(N):
            if matrix[col][row] == 0:
                if count == K:
                    total += 1
                count = 0
            else:
                count += 1
        if count == K:
            total += 1

    print("#{} {}".format(tc, total))