[Python] 백준 3029번 경고

문제

창영마을에서 정인이의 반란은 실패로 끝났다. (3028번)
테러리스트로 변신한 정인이는 창영마을에 경고를 하려고 한다.
사실 정인이는 창영마을에서 제일 착한사람이다. 따라서, 사람들을 다치지 않게하려고 한다.
유튜브에서 폭발에 대한 동영상을 찾아보다가, 그는 나트륨을 물에 던지면 폭발한다는 사실을 알게 되었다.

 

정인이는 창영마을의 중심을 지나는 "강산강" 근처에 숨어있다가, 나트륨을 위의 동영상처럼 물에 던질 것이다.
현재 시간과 정인이가 나트륨을 던질 시간이 주어졌을 때, 정인이가 얼마나 숨어있어야 하는지 구하는 프로그램을 작성하시오.

(정인이는 적어도 1초를 기다리며, 많아야 24시간을 기다린다.)

 

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

3029번: 경고

입력

첫째 줄에 현재 시간이 hh:mm:ss 형식으로 주어진다. 

(시, 분, 초) hh는 0보다 크거나 같고, 23보다 작거나 같으며, 분과 초는 0보다 크거나 같고, 59보다 작거나 같다.

둘째 줄에는 나트륨을 던질 시간이 위와 같은 형식으로 주어진다.

출력

첫째 줄에 정인이가 기다려야 하는 시간을 입력과 같은 형식으로 출력한다.

예제

# input
20:00:00
04:00:00
# output
08:00:00

# input
12:34:56
14:36:22
# output
02:01:26

문제 풀이

# 1
hour1, minute1, second1 = map(int, input().split(':'))
hour2, minute2, second2 = map(int, input().split(':'))
ans_h = ans_m = ans_s = 0

if hour1 == hour2 and minute1 == minute2 and second1 == second2:
    print('24:00:00')
else:
    if second2 - second1 < 0:
        minute2 -= 1
        ans_s = 60 + (second2 - second1)
    else:
        ans_s = second2 - second1
    
    if minute2 - minute1 < 0:
        hour2 -= 1
        ans_m = 60 + (minute2 - minute1)
    else:
        ans_m = minute2 - minute1
    
    if hour2 - hour1 < 0:
        ans_h = 24 + (hour2 - hour1)
    else:
        ans_h = hour2 - hour1

    print(f'{ans_h:02d}:{ans_m:02d}:{ans_s:02d}')

# 2
hour1, minute1, second1 = map(int, input().split(':'))
hour2, minute2, second2 = map(int, input().split(':'))
  
total_time1 = hour1 * 3600 + minute1 * 60 + second1
total_time2 = hour2 * 3600 + minute2 * 60 + second2

if total_time1 > total_time2:
    total_time2 += 86400

ans = total_time2 - total_time1
ans_h = ans // 3600
ans_m = ans % 3600 // 60
ans_s = ans % 60

if ans == 0:
    print('24:00:00')
else:
    print(f'{ans_h:02d}:{ans_m:02d}:{ans_s:02d}')

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,

혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.