[프로그래머스] 달리기 경주

Hyun·2023년 5월 20일
0

프로그래머스

목록 보기
1/32

달리기 경주

얀에서는 매년 달리기 경주가 열립니다. 해설진들은 선수들이 자기 바로 앞의 선수를 추월할 때 추월한 선수의 이름을 부릅니다. 예를 들어 1등부터 3등까지 "mumu", "soe", "poe" 선수들이 순서대로 달리고 있을 때, 해설진이 "soe"선수를 불렀다면 2등인 "soe" 선수가 1등인 "mumu" 선수를 추월했다는 것입니다. 즉 "soe" 선수가 1등, "mumu" 선수가 2등으로 바뀝니다.

<입출력 예>
players
["mumu", "soe", "poe", "kai", "mine"]
callings
["kai", "kai", "mine", "mine"]
result
["mumu", "kai", "mine", "soe", "poe"]

선수들의 이름이 1등부터 현재 등수 순서대로 담긴 문자열 배열 players와 해설진이 부른 이름을 담은 문자열 배열 callings가 매개변수로 주어질 때, 경주가 끝났을 때 선수들의 이름을 1등부터 등수 순서대로 배열에 담아 return 하는 solution 함수를 완성해주세요.

문제 핵심

선수A가 선수B를 추월할때 선수A(추월하는 선수)의 이름을 부름
이름이 불린 선수와 해당 선수의 바로 앞에 위치한 선수의 위치를 교환해준다.

방법1: 이중for문 이용 => 시간 초과 발생

  1. callings를 돌면서 해당 값이 players내 어디 위치해있는지 찾고
  2. 찾으면 players내 해당 값과 직전 값의 위치를 교환해준다.

def solution(players, callings):
    answer = []
    for c_index, c_value in enumerate(callings, start = 0):
        for p_index, p_value in enumerate(players, start = 0):
            if(c_value == p_value):
                #swap, 파이썬에서는 temp 변수 필요없음
                players[p_index - 1], players[p_index] = players[p_index], players[p_index - 1]
    
    answer = players
    
    return answer

방법2: 딕셔너리 이용

  1. callings돌면서 player_arr에서 이름에 해당하는 등수를 찾고
  2. index_arr 에서 해당 등수의 선수를 해당 등수 - 1의 선수와 위치를 swap 한다.
  3. player_arr에도 변경된 사항 적용해준다.

제출할땐 index_arr에서 value만 추출한걸 사용한다.

def solution(players, callings):
    answer = []
    
    #index: 현재 등수, player: 해당 등수의 선수이름
    index_arr = {index: player for index, player in enumerate(players)}
    #player: 선수이름, index: 해당 선수의 현재 등수
    player_arr = {player: index for index, player in enumerate(players)}
    
    for index, name in enumerate(callings):
        rank = player_arr[name]#불린 선수의 현재 등수
        
        #등수: 선수 딕셔너리에서 
        #rank-1: rank순위의 선수, rank: rank-1순위의 선수로 설정 
        index_arr[rank-1], index_arr[rank] = index_arr[rank], index_arr[rank-1]
        
        #선수: 등수 딕셔너리에서 (index_arr가 변경된 상태)
        #현재 불린선수: rank-1, 추월당한 선수(등수 내려옴, index_arr에 적용되어 있음): rank로 설정
        player_arr[name], player_arr[index_arr[rank]] = rank-1, rank 
        
    #딕셔너리에서 값만 추출해 배열로 만듦
    answer = list(index_arr.values())
    
    
    return answer
profile
better than yesterday

0개의 댓글