달리기 경주

프로그래머스 - 달리기 경주

문제

---

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

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

제한사항

---

• 5 ≤ players 길이 ≤ 50,000
  - players[i]는 i번째 선수의 이름
  • players의 원소들은 소문자로만 이루어져있다.
  • players 원소는 중복 X
  • 3 ≤ players[i] 길이 ≤ 10
• 2 ≤ callings 길이 ≤ 1,000,000
  - callings는 players의 원소들로만 이루어져있다.
  • 경주 진행중 1등인 선수의 이름은 불리지 않는다.

💡Idea(단순 swap)

1. 선수들의 등수를 어떻게 바꿀까?
   → players에서 불린 선수의 위치를 찾아 이전 선수와 바꾸자.

SourceCode

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

using namespace std;

vector<string> solution(vector<string> players, vector<string> callings) {
    for(int i=0; i<callings.size(); i++) {
        int idx = find(players.begin(), players.end(), callings[i]) - players.begin();
        swap(players[idx], players[idx-1]);
    }
   
    return players;
}

→ 시간 초과 발생,,

💡Idea(map)

1. find로 선수의 위치를 찾는 게 비효율적인 것 같다.
   → 선수 이름으로 접근 가능한 map을 사용하자 !
2. 선수의 등수는 어떻게 바꿀까?
   - 키는 선수 이름, 값은 등수로 가지는 map 하나만 생성하자
   → 추월한 선수의 이름을 찾을 수 없는 문제 발생
   - 추월한 선수의 등수를 바꾸기 위해서는 등수로 선수 이름을 찾을 수 있는 map이 필요
   → 등수와 선수 이름, 선수 이름과 등수를 저장하는 map을 두 개 생성하자 !

[ SourceCode ]

#include <string>
#include <vector>
#include <map>
#include <iostream>
#include <algorithm>

using namespace std;

vector<string> solution(vector<string> players, vector<string> callings) {
    vector <string> answer;
    map <string, int> rank; // 순위 찾기 위한 map
    map <int, string> name; // 이름 찾기 위한 map
    for(int i=0; i<players.size(); i++) {
        rank.insert({players[i], i+1}); // 선수 이름, 등수
        name.insert({i+1, players[i]}); // 등수, 선수 이름
    }
   
    for(int i=0; i<callings.size(); i++) {
        int idx = rank[callings[i]]; // 추월 전 등수
        string before = name[idx - 1]; // 추월할 선수
        rank[callings[i]]--;
        rank[before]++;
        name[idx] = before;
        name[idx - 1] = callings[i];
    }
   
    for (pair <int, string> str : name) {
        answer.push_back(str.second);
    }
   
    return answer;
}

Feedback

---

1. 처음에 map을 떠올렸으나 map을 두 개 생성할 생각을 하지 못함.
2. 단순 swap시 최악의 경우 callings의 길이가 1,000,000이고 players 크기가 50,000이다.
   → 시간 복잡도는 O(1000000 * 50000) 즉, O(50000000000)이다.
   → 1억 정도면 1초 정도의 시간이 걸린다.
   → 500억이면 500초가 걸린다고 생각하면 당연히 시간초과가 날 수 밖에 없다.
   ∴ 시간 복잡도를 계산해보는 습관 가지기 !!!