[백준] 기본 수학2 - 2563번 (Python)

https://www.acmicpc.net/problem/2563

2563번: 색종이

---

문제

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

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

입력

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

출력

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

코드 정답

num = int(input())                                      # 색종이의 수
paper = [[0 for _ in range(101)]for _ in range(101)]    # 2차원 배열

for _ in range(num):
    x, y = map(int, input().split())
    for i in range(x, x+10):                            # 가로가 10cm. 즉, 10칸
        for j in range(y, y+10):                        # 세로가 10cm. 즉, 10칸
            paper[i][j] = 1                             # 가로, 세로에 해당하는 사각형의 범위에 1 채워넣기.

ans = 0
for r in paper:                                         # 배열을 가져오기
    ans += r.count(1)                                   # 1이 존재하는 갯수 세어보기
print(ans)

풀이

1~2행 : 색종이의 수를 받고, 가로, 세로가 100인 정사각형 도화지를 생성한다.

num = int(input())                                      # 색종이의 수
paper = [[0 for _ in range(101)]for _ in range(101)]    # 2차원 배열

• 값 0을 넣어 빈 도화지로 만들어낸다.

 

4~8행 : 각각의 색종이의 가로, 세로를 입력받고 색종이가 존재하는 장소에 1을 넣어준다.

for _ in range(num):
    x, y = map(int, input().split())
    for i in range(x, x+10):                            # 가로가 10cm. 즉, 10칸
        for j in range(y, y+10):                        # 세로가 10cm. 즉, 10칸
            paper[i][j] = 1                             # 가로, 세로에 해당하는 사각형의 범위에 1 채워넣기

• 가로, 세로가 10cm 인 정사각형 색종이이기 때문에 +10을 해줘 10번씩 반복하게끔 한다.
• 배열 paper[i][j] 이 해당하는 장소에 값 1을 넣어줘 색칠해 준 것으로 만든다.
  두 색종이의 위치가 겹쳐도 값이 1인건 변하지 않는다.

 

10~13행 : 배열 paper 에 존재하는 값을 r 에 대입하며 1이 존재할 때마다 ans 에 1을 추가해준다.

ans = 0
for r in paper:                                         # 배열을 가져오기
    ans += r.count(1)                                   # 1이 존재하는 갯수 세어보기
print(ans)

• 배열을 1행 단위로 가져와 r 에 대입시킨 후 1이 존재하는 갯수 만큼 변수 ans 에 누적 형식으로 더해준다.