백준 2563번 '색종이' - [Python] 중복For, count

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

#️⃣ 2563번

문제

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

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

입력

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

출력

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

---

🔮 풀이

1. 흰색 도화지의 영역을 표현할 수 있는 리스트를 만든다
paper = [[0]*100 for i in range(100)]
→ 100*100 칸의 리스트를 0으로 초기화
2. 색종이의 수(n)과 색종이들의 좌표(a,b)를 입력받자

n = int(input)
for i in range(n) :
	a, b = map(int,input().split())

3. 입력받은 좌표로부터 10*10의 정사각형만큼 paepr리스트를 1로 바꿔준다

for i in range(a, a+10):
        for j in range(b,b+10):
            paper[i][j] = 1

이렇게 하면 겹치는 부분도 신경쓰지 않고 처리할 수 있다.
4. paper리스트에서 1의 개수를 세어주자
먼저 개수 셀때 효율을 위해서 최대로 채워진 x좌표를 구하고 여기까지만 셀 수 있도록 maxA를 설정해주자. 그리고 1의 개수를 새기위한 sumPaper도 정의한다.

maxA = 0
sumPaper =0

위의 for문에 maxA를 구하는 로직을 추가한다

for i in range(n):
    a, b = map(int, input().split())
    #추가된 부분
    if a>=maxA :
        maxA = a
    for i in range(a, a+10):
        for j in range(b,b+10):
            paper[i][j] = 1

python의 count를 사용해서 1의 개수를 세어보자

for i in range(maxA+10):
    sumPaper += paper[i].count(1)
print(sumPaper)

이렇게하면 겹친곳은 중복하지 않고 넓이를 구할 수 있다!

---

✨ 전체 코드

paper = [[0]*100 for i in range(100)]
n = int(input())
maxA = 0
sumPaper =0

for i in range(n):
    a, b = map(int, input().split())
    if a>=maxA :
        maxA = a
    for i in range(a, a+10):
        for j in range(b,b+10):
            paper[i][j] = 1

for i in range(maxA+10):
    sumPaper += paper[i].count(1)
print(sumPaper)