[Python/파이썬] 백준 20207번 달력

20207번: 달력

 

문제

수현이는 일년의 날짜가 1일부터 365일로 표시되어있는 달력을 가지고있다. 수현이는 너무나도 계획적인 사람이라 올 해 일정을 모두 계획해서 달력에 표시해놨다.

여름이 거의 끝나가자 장마가 시작되었고, 습기로 인해 달력에 표시한 일정이 지워지려고 한다. 지워지는 것을 막고자 수현이는 일정이 있는 곳에만 코팅지를 달력에 붙이려고 한다. 하지만 너무 귀찮았던 나머지, 다음과 같은 규칙을 따르기로 한다.

• 연속된 두 일자에 각각 일정이 1개 이상 있다면 이를 일정이 연속되었다고 표현한다.
• 연속된 모든 일정은 하나의 직사각형에 포함되어야 한다. 
• 연속된 일정을 모두 감싸는 가장 작은 직사각형의 크기만큼 코팅지를 오린다.

달력은 다음과 같은 규칙을 따른다.

• 일정은 시작날짜와 종료날짜를 포함한다.
• 시작일이 가장 앞선 일정부터 차례대로 채워진다.
• 시작일이 같을 경우 일정의 기간이 긴 것이 먼저 채워진다.
• 일정은 가능한 최 상단에 배치된다.
• 일정 하나의 세로의 길이는 1이다. 
• 하루의 폭은 1이다. 

 

위의 그림에서와 같이 일정이 주어졌다고 하자. 여기서 코팅지의 면적은 아래의 파란색 영역과 같다.

 

이때 코팅지의 크기의 합은 3 x 8 + 2 x 2 = 28이다. 

일정의 개수와 각 일정의 시작날짜, 종료날짜가 주어질 때 수현이가 자르는 코팅지의 면적을 구해보자.

입력

첫째 줄에 일정의 개수 N이 주어진다. (1 ≤ N ≤ 1000)

둘째 줄부터 일정의 개수만큼 시작 날짜 S와 종료 날짜 E가 주어진다. (1 ≤ S ≤ E ≤ 365)

출력

코팅지의 면적을 출력한다.

코드

n = int(input())
schedule = []
for _ in range(n):
    s, e = map(int, input().split())
    schedule.append([s, e])
    
schedule.sort(key=lambda x:(x[0], x[0]-x[1]))
calendar = [[False]*366 for _ in range(n)]
ans = 0
end_r = 0
start_c, end_c = 1, 0
# 달력에 일정 표시
for s, e in schedule:
    for i in range(n):
        if not calendar[i][s]:
            if i == 0:   # 첫째줄에 일정 들어가는 경우
                flag = False
                for j in range(n):  # 전날에 일정이 있었는지 체크
                    if calendar[j][s-1]:
                        flag = True
                        break
                if not flag:    # 새로운 시작 위치 설정
                    ans += (end_r+1) * (end_c-start_c+1)
                    start_c = s
                    end_c = e
                    end_r = 0
                else:           # 연속되게 일정이 있는 경우
                    end_r = max(end_r, i)
                    end_c = max(end_c, e)
            else:       # 첫째줄이 아닌 경우
                end_r = max(end_r, i)
                end_c = max(end_c, e)
            for j in range(s, e+1): # 일정 표시
                calendar[i][j] = True
            break    
print(ans+(end_r+1) * (end_c-start_c+1))

 

처음에 문제를 제대로 읽지 않아서 틀렸었다...

위와 같은 입력에 대해 처음에는 왼쪽 그림과 같이 코팅지를 붙여야 한다고 생각했는데 연속된 두 일자에 일정이 1개 이상 있다면 연속되었다고 본다고 했으므로 오른쪽과 같이 코팅지를 붙여야 한다.