[SWEA] 1948 날짜 계산기 Python

[ 문제 ]

 

난이도:  D2

문제 번호:  1948

 

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AV5PnnU6AOsDFAUq&categoryId=AV5PnnU6AOsDFAUq&categoryType=CODE&problemTitle=1948&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1

SW Expert Academy

---

 

월 일로 이루어진 날짜를 2개 입력 받아, 두 번째 날짜가 첫 번째 날짜의 며칠째인지 출력하는 프로그램을 작성하라.


[제약 사항]

월은 1 이상 12 이하의 정수이다. 각 달의 마지막 날짜는 다음과 같다.

1/31, 2/28, 3/31, 4/30, 5/31, 6/30, 7/31, 8/31, 9/30, 10/31, 11/30, 12/31

두 번째 날짜가 첫 번째 날짜보다 항상 크게 주어진다.

 

---

 

[ 코드 ]

월 일로 이루어진 날짜를 2개 입력 받아, 두 번째 날짜가 첫 번째 날짜의 며칠째인지 출력하는 프로그램을 작성하라.

T = int(input())
 
for test_case in range (1, T+1):
    numbers = list(map(int, input().split()))
    result = 0
    number = [1, 3, 5, 7, 8, 10, 12]
    if numbers[0] != numbers[2]:
        if numbers[0] in number:
            result += (32 - numbers[1]) + numbers[3]
        elif numbers[0] == 2:
            result += (29 - numbers[1]) + numbers[3]
        else:
            result += (31 - numbers[1]) + numbers[3]
        for i in range(numbers[0]+1, numbers[2]):
            if i in number:
                result += 31
            elif i == 2:
                result += 28
            else:
                result +=30
        print(f'#{test_case} {result}')
 
    else:
        result = numbers[3] - numbers[1] + 1
        print(f'#{test_case} {result}')