[백준] 2139번 나는 너가 살아온 날을 알고 있다 - 파이썬

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

2139번: 나는 너가 살아온 날을 알고 있다

풀이

일, 월, 연도를 받아준 뒤에 윤년인지 아닌지 확인한 뒤에 알맞는 날짜가 담긴 리스트를 가져가서 sum을 이용해서 합산해주면 됩니다.

# 2139번 나는 너가 살아온 날을 알고 있다
import sys
input = sys.stdin.readline
def leap(year):
    if year % 4 == 0:
        if year % 400 != 0 and year % 100 == 0:
            return [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
        else:
            return [0, 31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
    else:
        return [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]

while True:
    d, m, y = map(int, input().split())
    if (m,d,y) == (0,0,0):
        break
    li = leap(y)
    print(sum(li[:m])+d)

출력결과