문제
N X N 크기의 단어 퍼즐을 만들려고 한다. 입력으로 단어 퍼즐의 모양이 주어진다.
주어진 퍼즐 모양에서 특정 길이 K를 갖는 단어가 들어갈 수 있는 자리의 수를 출력하는 프로그램을 작성하라.
[예제]
N = 5, K = 3 이고, 퍼즐의 모양이 아래 그림과 같이 주어졌을 때
길이가 3 인 단어가 들어갈 수 있는 자리는 2 곳(가로 1번, 가로 4번)이 된다.
[제약 사항]
1. N은 5 이상 15 이하의 정수이다. (5 ≤ N ≤ 15)
2. K는 2 이상 N 이하의 정수이다. (2 ≤ K ≤ N)
[입력]
입력은 첫 줄에 총 테스트 케이스의 개수 T가 온다.
다음 줄부터 각 테스트 케이스가 주어진다.
테스트 케이스의 첫 번째 줄에는 단어 퍼즐의 가로, 세로 길이 N 과, 단어의 길이 K 가 주어진다.
테스트 케이스의 두 번째 줄부터 퍼즐의 모양이 2차원 정보로 주어진다.
퍼즐의 각 셀 중, 흰색 부분은 1, 검은색 부분은 0 으로 주어진다.
[출력]
테스트 케이스 t에 대한 결과는 “#t”을 찍고, 한 칸 띄고, 정답을 출력한다.
(t는 테스트 케이스의 번호를 의미하며 1부터 시작한다.)
작성 코드
import sys
sys.stdin = open("input.txt", "r")
T = int(input())
for test_case in range(1, T + 1):
N, K = map(int, input().split())
matrix = [[0] * (N) for _ in range(N)]
for i in range(N):
matrix[i] = list(map(int, input().split()))
ans = 0
for i in range(N):
cnt = [0,0] # 0:가로 1:세로
for j in range(N):
#가로
if matrix[i][j] == 0:
if cnt[0] == K:
ans += 1
cnt[0] = 0
else:
cnt[0] += 1
# 세로
if matrix[j][i] == 0:
if cnt[1] == K:
ans += 1
cnt[1] = 0
else:
cnt[1] += 1
if cnt[0] == K: ans += 1
if cnt[1] == K: ans += 1
print(f'#{test_case} {ans}')
'Problem Solving' 카테고리의 다른 글
[SWEA] D2 - 1970. 쉬운 거스름돈 in 파이썬 (0) | 2022.05.18 |
---|---|
[SWEA] D2 - 1989. 초심자의 회문 검사 in 파이썬 (0) | 2022.05.18 |
[SWEA] D2 - 2001. 파리 퇴치 in 파이썬 (0) | 2022.05.17 |
[SWEA] D2 - 1983. 조교의 성적 매기기 in 파이썬 (0) | 2022.05.17 |
[SWEA] D2 - 2005. 파스칼의 삼각형 in 파이썬 (0) | 2022.05.17 |