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

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

프로그래머스

문제 풀이

문제는 매우 간단하지만 단순히 반복문으로 플레이어 이름을 찾게되면 시간초과가 발생한다.

따라서 HashMap을 통해 이름과 순위를 저장해 빠르게 선수를 찾아 순위를 업데이트 하도록 구현.

 

코드

import java.util.*;
class Solution {
    public String[] solution(String[] players, String[] callings) {
        //       이름      순위
        HashMap<String, Integer> p_rank = new HashMap<>();
        
        for(int i=0; i<players.length; i++){
            // 순위는 1부터 시작
            p_rank.put(players[i], i+1);
        }
        
        for(int i=0; i<callings.length; i++){
	        // 호출된 사람의 순위를 감소
            String callP=callings[i];
            int rank = p_rank.get(callP);
            p_rank.put(callP, rank-1);
            
            // 호출된 사람 앞사람의 순위를 증가
            String frontP = players[rank-1-1];
            p_rank.put(frontP, rank);
            
            // 바뀐 순서에 맞게 players 업데이트
            players[rank-2]=callP;
            players[rank-1]=frontP;
        }
        return players;
        
    }
}

 

rank는 1부터 시작하기 때문에 players에서 선수를 조회할 때는 -1이 디폴트이고

앞선수를 조회할 때 추가로 -1이 더 들어가기 때문에 -2가 나온다