[프로그래머스] 달리기 경주 - 파이썬

Programmers

달리기 경주

https://school.programmers.co.kr/learn/courses/30/lessons/178871

프로그래머스

문제

코드

• 시간 초과 코드

def solution(players, callings):

    for i in callings:
        n = players.index(i)
        players[n], players[n-1] = players[n-1], players[n]
    return players

 

• dictionary, enumerate를 활용하여 시간복잡도 줄이기
• 현재 선수의 등수 idx를 저장

def solution(players, callings):
    lst = {player: i for i, player in enumerate(players)} 

    for j in callings:
        idx = lst[j] 
        lst[j] -= 1 
        lst[players[idx-1]] += 1 
        players[idx-1], players[idx] = players[idx], players[idx-1] 

    return players