프로그래머스 - [PCCP 기출문제] 1번 / 동영상 재생기 (Python)

Lv1, 문자열, 구현

문제

당신은 동영상 재생기를 만들고 있습니다. 당신의 동영상 재생기는 10초 전으로 이동, 10초 후로 이동, 오프닝 건너뛰기 3가지 기능을 지원합니다. 각 기능이 수행하는 작업은 다음과 같습니다.

• 10초 전으로 이동: 사용자가 "prev" 명령을 입력할 경우 동영상의 재생 위치를 현재 위치에서 10초 전으로 이동합니다. 현재 위치가 10초 미만인 경우 영상의 처음 위치로 이동합니다. 영상의 처음 위치는 0분 0초입니다.
• 10초 후로 이동: 사용자가 "next" 명령을 입력할 경우 동영상의 재생 위치를 현재 위치에서 10초 후로 이동합니다. 동영상의 남은 시간이 10초 미만일 경우 영상의 마지막 위치로 이동합니다. 영상의 마지막 위치는 동영상의 길이와 같습니다.
• 오프닝 건너뛰기: 현재 재생 위치가 오프닝 구간(op_start ≤ 현재 재생 위치 ≤ op_end)인 경우 자동으로 오프닝이 끝나는 위치로 이동합니다.

동영상의 길이를 나타내는 문자열 video_len, 기능이 수행되기 직전의 재생위치를 나타내는 문자열 pos, 오프닝 시작 시각을 나타내는 문자열 op_start, 오프닝이 끝나는 시각을 나타내는 문자열 op_end, 사용자의 입력을 나타내는 1차원 문자열 배열 commands가 매개변수로 주어집니다. 이때 사용자의 입력이 모두 끝난 후 동영상의 위치를 "mm:ss" 형식으로 return 하도록 solution 함수를 완성해 주세요.

풀이

1. 내가 일단 작성한 풀이

• 풀이 자체는 매우 쉬웠다.
  오랜만에 format()함수를 써서 {:02d}하는게 가장 큰 난관이었음

def solution(video_len, pos, op_start, op_end, commands):
    # 동영상 길이, 직전 재생위치, 오프닝 시작, 끝, 사용자 입력
    # 사용자 입력 끝난 후의 동영상 위치 리턴
    
    # next : +10s
    # prev : -10s
    end = int(video_len.split(':')[0]) * 60 + int(video_len.split(':')[1])
    op_st = int(op_start.split(':')[0]) * 60 + int(op_start.split(':')[1])
    op_ed = int(op_end.split(':')[0]) * 60 + int(op_end.split(':')[1])
    
    now_min = int(pos.split(':')[0])
    now_sec = int(pos.split(':')[1])
    
    now = now_min * 60 + now_sec
    for com in commands:
        if op_st <= now <= op_ed:
            now = op_ed
        
        if com == 'prev':
            # -10
            if now - 10 >= 0:
                now -= 10
            else:
                now = 0
        else:
            # +10
            if now + 10 <= end:
                now += 10
            else:
                now = end
    
    if op_st <= now <= op_ed:
        now = op_ed
    
    ans_min = now // 60
    ans_sec = now % 60
    answer = "{:02d}:{:02d}".format(ans_min, ans_sec)
    return answer

2. Cursor.ai가 리팩토링한 코드

• 전체적인 로직은 문제가 워낙 쉽다보니 변화가 있진 않다.
  다만, 반복적으로 계산되는 min -> sec의 과정을 함수로 변경해 가독성 향상 및 코드 중복을 감소시켰다.
• 또, 기존 if문으로 확인하던 부분을 min(), max() 함수로 변경한 것도 내가 놓친 부분이었다.
  대충 사용한 AI 툴 기능이 이런데, 누가 직접 처음부터 코드 다 짜나 싶다...

def time_to_seconds(time_str):
    minutes, seconds = map(int, time_str.split(':'))
    return minutes * 60 + seconds

def seconds_to_time(seconds):
    return "{:02d}:{:02d}".format(seconds // 60, seconds % 60)


def solution(video_len, pos, op_start, op_end, commands):
    # 동영상 길이, 직전 재생위치, 오프닝 시작, 끝, 사용자 입력
    # 사용자 입력 끝난 후의 동영상 위치 리턴
    
    # next : +10s
    # prev : -10s
    video_end = time_to_seconds(video_len)
    op_st = time_to_seconds(op_start)
    op_ed = time_to_seconds(op_end)
    
    now = time_to_seconds(pos)
    for command in commands:
        if op_st <= now <= op_ed:
            now = op_ed
        
        if command == 'prev':
            # -10
            now = max(0, now - 10)
        else:
            # +10
            now = min(video_end, now + 10)
    
    if op_st <= now <= op_ed:
        now = op_ed
    
    return seconds_to_time(now)