문제

연도와 월을 매개변수로 입력받아, 그 달이 며칠까지 있는지 반환하는 함수 days_in_month()를 작성하라. 이때, 2월의 길이는 윤년인지 아닌지에 따라 다르다. 윤년 계산을 위해 연습문제 6-2에서 만든 함수를 활용하라.

답

코드

def is_leap_year(year):
    """연도를 입력받아 그 해가 윤년인지를 반환한다."""
    return (year % 4 == 0) and (not (year % 100 == 0)) or (year % 400 == 0)

# 날이 31개 있는 달의 집합
months_with_31_days = {1, 3, 5, 7, 8, 10, 12}

# 날이 30개 있는 달의 집합
months_with_30_days = {4, 6, 9, 11}

# 날이 28개 또는 29개 있는 달의 집합
months_with_28_or_29_days = {2}

def days_in_month(year, month):
    """연도와 월을 입력받아 그 달이 며칠까지 있는지 반환한다."""
    if month in months_with_31_days:
        return 31

    if month in months_with_30_days:
        return 30

    if month in months_with_28_or_29_days:
        return 29 if is_leap_year(year) else 28

print('2020년 2월의 날의 수:', days_in_month(2020, 2))
print('2021년 2월의 날의 수:', days_in_month(2021, 2))
print('2021년 5월의 날의 수:', days_in_month(2020, 5))
print('2021년 9월의 날의 수:', days_in_month(2020, 11))

실행 결과

2020년 2월의 날의 수: 29
2021년 2월의 날의 수: 28
2021년 5월의 날의 수: 31
2021년 9월의 날의 수: 30

해설

• if 문으로 달이 며칠까지 있는 달인지를 확인하여 그에 맞는 수를 반환하면 된다.
• 이 때, 각 달을 if month == 1:, if month == 2:, if month == 3:... 과 같이 나열하는 방식도 한 방법이다. 쉬운 방법이지만 코드가 너무 길어지는 단점이 있다.
• 답안에서는 5장에서 다룬 집합을 이용해 '날이 31개 있는 달의 집합', '날이 30개 있는 달의 집합', '날이 28개 또는 29개 있는 달의 집합'을 각각 정의해두고, 함수에서 이 집합을 활용해 달을 검사하도록 했다.
• 달이 2월인 경우에는 연도의 윤년 여부에 따라 29 또는 28을 반환해야 한다. 풀이에서는 코드를 간결하게 작성하려고 조건부 식을 활용했다. 물론 그냥 if 문을 사용해도 된다.