[SWEA/d2] 1948. 날짜 계산기 (Python)

📚 문제 링크

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

 

---

 

📒 코드

풀이 1. datetime() 사용하기

from datetime import datetime

T = int(input())
for tc in range(1, T+1):
    m1, d1, m2, d2 = map(int, input().split())
    
    day1 = datetime(2023, m1, d1)
    day2 = datetime(2023, m2, d2)
    
    diff = day2 - day1

    print("#{} {}" .format(tc, diff.days + 1))

 

 

풀이 2. 월별로 날짜가 누적된 리스트를 만들기

month = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31]
accum_month = [] # 누적 일수
result = 0

for i in month:
    result += i
    accum_month.append(result)
    # accum_month = [0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 365]

T = int(input())
for tc in range(1, T+1):
    m1, d1, m2, d2 = map(int, input().split())
    # m1, d1이랑 m2, d2랑 1월1일부터의 각각의 날짜를 계산하고 뺀 뒤에 +1
    result = (d2 + accum_month[m2-1]) - (d1 + accum_month[m1-1]) + 1

    print("#{} {}" .format(tc, result))

 

날짜가 누적된 달을 리스트로 만들고, 누적된 날짜를 계산해 준다.

(d2 + accum_month[m2-1]) : m2월 d2일까지의 누적된 날짜 계산

(d1 + accum_month[m1-1]) : m1월 d1일까지의 누적된 날짜 계산

 

 

---

 

💡 새롭게 알게 된 지식

• datetime(year, month, day)
• datetime 형식의 두 날짜를 뺀 후, .days 하면 일수 차이가 나온다 !
• 누적달 구하기

 

---

 

🔎 결과

풀이 1

풀이 2

 

datetime()을 사용한 것보다, 누적달을 계산해서 풀이한 쪽이 시간/메모리 측면에서 조금 더 나은 것을 알 수 있다.