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

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=2&contestProbId=AV5PuPq6AaQDFAUq&categoryId=AV5PuPq6AaQDFAUq&categoryType=CODE&problemTitle=&orderBy=PASS_RATE&selectCodeLang=PYTHON&select-1=2&pageSize=10&pageIndex=1 

SW Expert Academy

 

문제

N X N 크기의 단어 퍼즐을 만들려고 한다. 입력으로 단어 퍼즐의 모양이 주어진다.

주어진 퍼즐 모양에서 특정 길이 K를 갖는 단어가 들어갈 수 있는 자리의 수를 출력하는 프로그램을 작성하라.

 

제출 답안 & 참고 답안

T = int(input())
for test_case in range(1, T + 1):
      n, k = map(int, input().split())
      board = [list(map(int, input().split())) for _ in range(n)]       # 2차원 배열로 만들기   

      answer = 0
      for i in range(n):
            cnt = 0
            for j in range(n):
                  if board[i][j] == 1:
                        cnt += 1
                  if board[i][j] == 0 or j==n-1:
                        if cnt == k:
                              answer += 1
                        cnt = 0
            for j in range(n):
                  if board[j][i] == 1:
                        cnt += 1
                  if board[j][i] == 0 or j==n-1:
                        if cnt == k:
                              answer += 1
                        cnt = 0    
      print(f"#{test_case} {answer}")

 

참고 자료

https://spongebob88.tistory.com/65

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