[Python | 백준 1308번] D-Day

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

1308번: D-Day

 

문제

캠프에 오게 된 송유진은 캠프가 너무 지루해서 오늘로부터 캠프가 끝날 때 까지 며칠이나 남았는지 알아보고 싶었다. 그런데 캠프는 비상식적으로 길지도 몰라서 (윤년을 포함할지도 모른다) 손으로 하나하나 세기에는 힘들어 보였다.

더욱 정확한 계산을 위해, 유진이는 윤년이 정해지는 기준을 찾아보았고, 그것은 다음과 같았다.

• 서력기원 연수가 4로 나누어떨어지는 해는 우선 윤년으로 한다. (2004년, 2008년, …)
• 100으로 나누어떨어지는 해는 평년으로 한다. (2100년, 2200년, …)
• 400으로 나누어떨어지는 해는 다시 윤년으로 한다. (1600년, 2000년, …)

그런데 캠프가 너무 길 경우, 사춘기인 유진이는 캠프에 무단으로 빠질지도 모른다.

입력

첫째 줄에 오늘의 날짜가 주어지고, 두 번째 줄에 D-Day인 날의 날짜가 주어진다. 날짜는 연도, 월, 일순으로 주어지며, 공백으로 구분한다. 입력 범위는 1년 1월 1일부터 9999년 12월 31일 까지 이다. 오늘의 날짜는 항상 D-Day보다 앞에 있다.

출력

오늘부터 D-Day까지 x일이 남았다면, "D-"를 출력하고 그 뒤에 공백 없이 x를 출력한다. 만약 캠프가 천년 이상 지속된다면 (오늘이 y년 m월 d일이고, D-Day가 y+1000년 m월 d일과 같거나 늦다면) 대신 "gg"를 출력한다. 오늘이 2월 29일인 경우는 주어지지 않는다.

 

---

내 풀이

# 38072kb 68ms

from datetime import date
y1, m1, d1 = map(int, input().split())
y2, m2, d2 = map(int, input().split())

today = date(y1, m1, d1)
end_date = date(y2, m2, d2)
day_long = end_date - today
d_day = day_long.days # 애초에 윤년도 계산해주는 듯함

if (y1+1000) < y2:      # 1-1)천년 이상 차이
  print('gg')
elif y1+1000 == y2:     # 1-2)딱 천년 차이
  if m1 > m2:               # 2-1) 월 차이: 천년 미만
    print(f"D-{d_day}")
  elif m1 == m2:            # 2-2) 월이 같으면 일수 확인
    if d1 > d2 :
      print(f"D-{d_day}")       # 3-1) 일 차이: 천년 미만
    else:
      print("gg")               # 3-2) 일 차이: 천년 이상
  else:                     # 2-3) 월 차이 : 천년 이상
    print('gg')
else:                   # 1-3) 천년 미만
  print(f"D-{d_day}")

 

메모

원래는 윤년도 계산해서 처리해야 하는데

datetime 패키지를 가져와서 날짜 차이를 구하면 윤년까지 고려해서 날짜 차이를 계산해준다.이 때문에 출력하는 과정에서 천년 이상에 대한 if문만 추가로 작성하면 되어서 문제가 한결 쉬워졌다.