[백준] 파이썬/ 10163 색종이

📌 10163 색종이

 

내 풀이

n = int(input())
paper = []

for i in range(n):
    paper.append(list(map(int,input().split())))

place = [[0]*1001 for _ in range(1001)]

for i,p in enumerate(paper):
    cnt=0
    [row,col,area,height] = p
    for r in range(row,row+area):
        for c in range(col,col+height):
            place[r][c]=i+1

for i in range(n):
    cnt=0
    for p in place:
        cnt += p.count(i+1)
    print(cnt)

 

단적인 예로

2

0 0 10 10

2 2  6  6 

을 입력했을 시 

첫번째  색종이는 1로 채워지고

두번째 색종이는 2로 채워져 1이 덮어지는 구조로

 

마지막에 각 idx의 값을 셈으로써 색종이가 차지하는 넓이를 알아내는 방식이다.

 

 

 

 

 

 

근데 여기서 시간초과가 난다.

그 이유는 바로 각 idx를 채우는 이중for 문때문 

# 변경 전 이중 for문
for r in range(row,row+area):
	for c in range(col,col+height):
    		place[r][c]=i+1
        

# 변경 후 
for r in range(row,row+area):
	place[r][col:col+height] = [i+1]*height

변경 전 

[1,1,1,1,1,1] -> [2,1,1,1,1,1]   X6번 작업

[1,1,1,1,1,1] -> [2,2,2,2,2,2]  X1번 작업 으로 변경 

 

변경 전에는 이중 for 문을 돌면서 저 값을 채워주지만

변경 후에는  행 전체를 통째로 만들어 교체함으로써 for문을 한번만 돌게 되었다.

 

 

 

개선 풀이

n = int(input())
paper = []

for i in range(n):
    paper.append(list(map(int,input().split())))

place = [[0]*10 for _ in range(10)]

for i,p in enumerate(paper):
    cnt=0
    [row,col,area,height] = p
    
    for r in range(row,row+area):
        place[r][col:col+height] = [i+1]*height

for i in range(n):
    cnt=0
    for p in place:
        cnt += p.count(i+1)
    print(cnt)