[코딩 테스트 연습(파이썬/Python)] 백준(BOJ) 2457번 _ 공주님의 정원

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

2457번: 공주님의 정원

[문제]

오늘은 공주님이 태어난 경사스러운 날이다. 왕은 이 날을 기념하기 위해 늘 꽃이 피어있는 작은 정원을 만들기로 결정했다.

총 N개의 꽃이 있는 데, 꽃은 모두 같은 해에 피어서 같은 해에 진다. 하나의 꽃은 피는 날과 지는 날이 정해져 있다. 예를 들어, 5월 8일 피어서 6월 13일 지는 꽃은 5월 8일부터 6월 12일까지는 꽃이 피어 있고, 6월 13일을 포함하여 이후로는 꽃을 볼 수 없다는 의미이다. (올해는 4, 6, 9, 11월은 30일까지 있고, 1, 3, 5, 7, 8, 10, 12월은 31일까지 있으며, 2월은 28일까지만 있다.)

이러한 N개의 꽃들 중에서 다음의 두 조건을 만족하는 꽃들을 선택하고 싶다.

1. 공주가 가장 좋아하는 계절인 3월 1일부터 11월 30일까지 매일 꽃이 한 가지 이상 피어 있도록 한다.
2. 정원이 넓지 않으므로 정원에 심는 꽃들의 수를 가능한 적게 한다. 

N개의 꽃들 중에서 위의 두 조건을 만족하는, 즉 3월 1일부터 11월 30일까지 매일 꽃이 한 가지 이상 피어 있도록 꽃들을 선택할 때, 선택한 꽃들의 최소 개수를 출력하는 프로그램을 작성하시오. 

 

[입력]

첫째 줄에는 꽃들의 총 개수 N (1 ≤ N ≤ 100,000)이 주어진다. 다음 N개의 줄에는 각 꽃이 피는 날짜와 지는 날짜가 주어진다. 하나의 날짜는 월과 일을 나타내는 두 숫자로 표현된다. 예를 들어서, 3 8 7 31은 꽃이 3월 8일에 피어서 7월 31일에 진다는 것을 나타낸다. 

[출력]

첫째 줄에 선택한 꽃들의 최소 개수를 출력한다. 만약 두 조건을 만족하는 꽃들을 선택할 수 없다면 0을 출력한다.

 

[아이디어]

1) 3월 1일(s) 보다 먼저 피면서 가장 늦게까지 피는 꽃을 찾는다.

2) 1)의 꽃의 지는 날로 s를 업데이트 한다.

3) 1)을 반복한다.

4) 문제를 잘 읽자.... 지는 날은 꽃이 피어있지 않고, 0을 출력해야 하는 경우도 존재할 수 있다.

 

import sys
input=sys.stdin.readline
from itertools import accumulate
n=int(input())
day = list(accumulate([0,31,28,31,30,31,30,31,31,30,31,30,31]))
f = []
s = day[2]+1
e = day[11]
for _ in range(n):
    a,b,c,d = map(int,input().split())
    f.append((day[a-1]+b,day[c-1]+d))
f.sort(key=lambda x:x[1],reverse=True)
cnt = 0
while s<=e:
    check=False
    for x,y in f:
        if x<=s and y>m:
            m=y
            cnt+=1
            check=True
            break
    s=m
    if not check:
        cnt=0
        break
print(cnt)

from itertools import accumulate
n=int(input())
day = list(accumulate([0,31,28,31,30,31,30,31,31,30,31,30,31]))
f = []
s = day[2]+1
e = day[11]
for _ in range(n):
    a,b,c,d = map(int,input().split())
    f.append((day[a-1]+b,day[c-1]+d))
f.sort(key=lambda x:x[0])
cnt = 0
i=0
m=0
while s<=e:
    check=False
    for x in range(i,n):
        if f[x][0]<=s:
            if m<f[x][1]:
                m=f[x][1]
                check = True
        elif f[x][0]>s:
            break
    if check:
        i=x
        s=m
        cnt+=1
    else:
        cnt=0
        break
print(cnt)

처음에는 filter를 활용한 접근을 했는데, 계속해서 시간 초과가 나는 것이 문제를 대충 읽어서가 아니라 시간 복잡도 상의 문제라고만 생각해서, 한참을 헤맸다.

import sys
i=sys.stdin.readline
p={1:0,2:31,3:59,4:90,5:120,6:151,7:181,8:212,9:243,10:273,11:304,12:334}
f=[];s=p[3]+1;e=p[12]
for _ in range(int(i())):
    a,b,c,d=map(int,i().split())
    f.append((p[a]+b,p[c]+d))
f.sort(key=lambda x:x[1]);t=0
while s<=e:
    k=list(filter(lambda x:x[0]<=s,f))
    if k and s<k[-1][1]:s=k[-1][1];t+=1
    else:t=0;break
print(t)

헤맨게 아까워서 처음 작성했던 코드 기반으로 숏코딩을 하나 만들었다. 효율은 당연히 구리다. Pypy3로 시간초과가 나는데 파이썬으로는 통과되는게 신기했다.