[Python/BOJ] 20207. 달력

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

20207번: 달력

문제

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

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

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

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

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

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

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

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

입력

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

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

출력

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

풀이

Code

import sys
input = sys.stdin.readline

# 1. check 재귀 함수 선언
def check(s, e) :
    # 1-1. 일 별 일정 수 카운트
    for i in range(s, e+1) :
        schedule[i] += 1

n = int(input())
# 2. 주어진 일정 정렬
array = sorted([list(map(int, input().split())) for _ in range(n)], key = lambda x : [x[0], -x[1]])
# 3. 일정 리스트 생성
schedule = [0] * 366
answer = 0

# 4. start, end 초기 설정
start, end = array[0][0], array[0][1]
# 5. 일정 카운트
check(array[0][0], array[0][1])
# 6. for 문
for s, e in array[1:] :
    # 6-1. 일정이 연속될 경우
    if end + 1 >= s :
        # end 재설정
        if e > end: end = e
    # 6-2. 일정이 연속되지 않을 경우
    else :
        # 해당 코팅지 면적 더하기
        answer += (end - start + 1) * max(schedule[start:end+1])
        start, end = s, e
    # 7. 일정 카운트
    check(s, e)
# 8. 가장 마지막 연속된 일정 처리
else : answer += (end - start + 1) * max(schedule[start:end+1])
# 9. 결과 출력
print(answer)