[python] 백준 - 1780. 종이의 개수

🤔문제 해결

• S2 | 분할정복

첫번째에 있는 코드가 더 정답인거 같다( 둘다 통과했지만 )

첫번째 코드는 항상 9개의 조각으로 잘라서 진행한다. - 이게 문제랑 더 맞는거 같다.

하지만

두번째 코드는 (처음 풀어서 통과하고 다시보니 이상함) 9개의 조각이 아니다.

종이가 들어오면 3x3으로 다 잘라버린다.

그래도 통과한다.

이게 되네??

내가 문제를 잘 이해하지 못한건지... 내가 짠 코드를 이해하지 못한건지 잘 모르겠다.

💻소스 코드

from itertools import chain  # 2차원 리스트를 1차원리스트로 바꿔주는 친구


def bt(paper):
    global a, b, c
    nn = len(paper)

    if len(set(chain.from_iterable(paper))) == 1:
        if paper[0][0] == -1:
            a += 1
        elif paper[0][0] == 0:
            b += 1
        else:
            c += 1
    else:
        if nn != 3:
            for i in range(0, nn, nn//3):
                for j in range(0, nn, nn//3):
                    tmp_paper = [row[j:j + nn//3] for row in paper[i:i + nn//3]]
                    bt(tmp_paper)
        else:
            for i in range(nn):
                for j in range(nn):
                    if paper[i][j] == -1:
                        a += 1
                    elif paper[i][j] == 0:
                        b += 1
                    else:
                        c += 1


n = int(input())
papers = [list(map(int, input().split())) for _ in range(n)]
a, b, c = 0, 0, 0
bt(papers)
print(f'{a}\n{b}\n{c}')

from itertools import chain


def bt(paper):
    global a, b, c
    nn = len(paper)

    if len(set(chain.from_iterable(paper))) == 1:
        if paper[0][0] == -1:
            a += 1
        elif paper[0][0] == 0:
            b += 1
        else:
            c += 1
    else:
        if nn != 3:
            for i in range(0, nn, 3):
                for j in range(0, nn, 3):
                    tmp_paper = [row[j:j + 3] for row in paper[i:i + 3]]
                    bt(tmp_paper)
        else:
            for i in range(nn):
                for j in range(nn):
                    if paper[i][j] == -1:
                        a += 1
                    elif paper[i][j] == 0:
                        b += 1
                    else:
                        c += 1


n = int(input())
papers = [list(map(int, input().split())) for _ in range(n)]
a, b, c = 0, 0, 0
bt(papers)
print(a)
print(b)
print(c)

📕문제 확인

출처: BACKJOON ONLINE JUDGE

링크: https://www.acmicpc.net/problem/1780

1780번: 종이의 개수