[자바] 달리기 경주

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

프로그래머스

 

문제설명

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

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

 

나의 문제 풀이

1)  for문을 써서 문제를 풀었으나 테스트 11번 부터 시간초과가 걸려서 실패했기 때문에 고민하다 HashMap을 선택했다.

2) 1번째 for문에서 players의 이름과 등수를 HashMap에 담아준다.

3) 2번째 for문에서는 callings에서 이름을 불린 선수 p를 hashMap에 있는 선수와 비교하여 등수를 가져오고 rank에 담는다.

4) 예를 들어 rank가 2라면 fp에는 2보다 앞에 있는 1인 선수를 담아주고 replace를 통해 rank가 1인 선수를 2로 바꿔주고 

2인 선수는 1이 되게 한다. 쉽게 말하면 앞에 있는 선수와 등수를 바꿔준다.

 

import java.util.*;
class Solution {
    public String[] solution(String[] players, String[] callings) {
        HashMap<String, Integer> hashMap = new HashMap<>();
        for(int i = 0; i < players.length; i++){
            hashMap.put(players[i], i);
        }
        for(String p : callings){
            int rank = hashMap.get(p);

            String fp = players[rank-1];
            hashMap.replace(fp, rank);
            players[rank] = fp;

            hashMap.replace(p, rank-1);
            players[rank-1] = p;
        }
        
        return players;
    }
}