[백준] 2563: 색종이 (Python)

문제

가로, 세로의 크기가 각각 100인 정사각형 모양의 흰색 도화지가 있다. 이 도화지 위에 가로, 세로의 크기가 각각 10인 정사각형 모양의 검은색 색종이를 색종이의 변과 도화지의 변이 평행하도록 붙인다. 이러한 방식으로 색종이를 한 장 또는 여러 장 붙인 후 색종이가 붙은 검은 영역의 넓이를 구하는 프로그램을 작성하시오.

 

예를 들어 흰색 도화지 위에 세 장의 검은색 색종이를 그림과 같은 모양으로 붙였다면 검은색 영역의 넓이는 260이 된다.

입력

첫째 줄에 색종이의 수가 주어진다. 이어 둘째 줄부터 한 줄에 하나씩 색종이를 붙인 위치가 주어진다. 색종이를 붙인 위치는 두 개의 자연수로 주어지는데 첫 번째 자연수는 색종이의 왼쪽 변과 도화지의 왼쪽 변 사이의 거리이고, 두 번째 자연수는 색종이의 아래쪽 변과 도화지의 아래쪽 변 사이의 거리이다. 색종이의 수는 100 이하이며, 색종이가 도화지 밖으로 나가는 경우는 없다

출력

첫째 줄에 색종이가 붙은 검은 영역의 넓이를 출력한다.

 

풀이

솔직히 아무 정보 없이 문제를 봤다면 해결방법을 못 찾았을 수도 있을 것 같다.. 그런데 2차원 배열 카테고리에 있었기 때문에 보다 쉽게 방법을 찾을 수 있었다.

모든 원소가 0인 전체 넓이 배열을 만들어주고 입력받은 위치부터 +10까지의 원소를 1로 바꾸고 1의 총 개수를 출력하면 된다.

처음에는 마지막 위치까지 넓이에 포함되는 줄 알았는데, 마지막 위치 (ex. 3 7 을 입력받았을 경우 13 17) 는 넓이에 포함되지 않는다.

 

그리고 처음에 배열을 선언할 때 [[0] * [100]] * 100으로 했는데, 이렇게 하면 같은 리스트를 복제한 것이라 같은 객체를 100번 참조하게 되어서 본래 의도하던 바와 달라지므로 수정했다.

 

정답 코드

import sys
input = sys.stdin.readline

arr = [[0] * 100 for i in range(100)] # 전체 넗이
p = int(input())

for _ in range(p):
    a, b = map(int, input().split())
    for i in range(a, a+10):
        for j in range(b, b+10):
            arr[i][j] = 1
ans = 0
for i in range(100):
    ans += arr[i].count(1)

print(ans)

 

알게 된 점

• 다차원 배열 선언하는 방법