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

SW Expert Academy

---

🗒️파이썬 코드 풀이

T = int(input())
for tc in range(1,T+1):
    N,K = map(int,input().split())
    lst = [list(map(int,input().split())) + [0] for _ in range(N)]
    lst.append([0]*(N+1))
    ans = 0

    v = []
    for i in range(N+1):
        sum = 0
        for j in range(N+1):
            if lst[i][j] == 1 :
                sum += 1
            elif lst[i][j] == 0 :
                v. append(sum)
                sum = 0
    ans += v.count(K)

    v = []
    for j in range(N+1):
        for i in range(N+1):
            if lst[i][j] == 1:
                sum += 1
            elif lst[i][j] == 0:
                v.append(sum)
                sum = 0
    ans += v.count(K)
    print(f"#{tc} {ans}")

 

 

1. 기존 입력받은 리스트에 N+1 행과 열을 0으로 채워준다.

 

2. 먼저 행을 진행해주는데, lst[i][j]가 1일때는 계속 더해주고, 0일때는 v 리스트에 append하고 초기화 한다.

 

3. 행에서 sum 값들을 append 한 후에 count()를 통해 원하는 값의 개수를 세어주고 ans에 더해준다.

 

4. 열도 마찬가지로 진행 해주면 된다. ( i와 j의 반복문 위치만 바꿔주면 됨)

 

📌  문제 코멘트

di,dj 방식으로 푼적 있지만, 꽤나 복잡했다.

이런것은 크게 방향이 왔다갔다 하지 않으므로, 반복문으로 충분히 가능하다. 

또한, 0을 패딩해주는 방법도 있다는 것을 꼭 기억하자 

---

📚문제

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부터 시작한다.)