본문 바로가기

Algorithm/Programers - C++

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

 

 

Level. 1

 

문제

얀에서는 매년 달리기 경주가 열립니다.
해설진들은 선수들이 자기 바로 앞의 선수를 추월할 때 추월한 선수의 이름을 부릅니다.
예를 들어 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에는 중복된 값이 들어가 있지 않습니다.
   3 ≤ players[i]의 길이 ≤ 10
- 2 ≤ callings의 길이 ≤ 1,000,000
   callings는 players의 원소들로만 이루어져 있습니다.
   경주 진행중 1등인 선수의 이름은 불리지 않습니다.

 

풀이

#include <string>
#include <vector>
#include <algorithm>
#include <map>
#include <iostream>
using namespace std;

vector<string> solution(vector<string> players, vector<string> callings) {
    vector<string> answer = players;
    map<string, int> names;
    
    /*
    
    // index를 구하는 부분에서 효율성 통과를 하지 못한다. 
    // map 자료구조를 이용하도록 한다. 
    
    vector<string> answer = players;
    for(int i=0; i<callings.size(); i++){
        int index = find(answer.begin(), answer.end(), callings[i]) - answer.begin(); 
        string temp;
        temp = answer[index-1];
        answer[index-1] = answer[index];
        answer[index] = temp;
    }
    */
    
    for(int i=0; i<answer.size(); i++){
        names[answer[i]] = i;
    }
    
    for(int i=0; i<callings.size(); i++){
        int lank = names[callings[i]];
        names[callings[i]] = lank - 1;
        names[answer[lank-1]] = lank;
        string temp = answer[lank-1];
        answer[lank-1] = answer[lank];
        answer[lank] = temp;
        
    }
    return answer;
}

 

사용한 자료구조

  • map : key값으로 선수이름을, value값으로 해당 선수의 현재 순위를 저장하기 위해 사용했다. vector의 find를 사용해 index를 구하기엔 효율성 테스트에서 통과할 수 없었기 때문이다. 

해결방법

  1. answer 벡터에는 key로 순위를, value로 선수이름을 저장했다. 
  2. names 에는 key값으로 선수이름을, value값으로 해당 선수의 현재 순위를 저장했다.
  3. callings 벡터를 순회하면서 해당 선수의 순위와 바로 앞 선수의 순위를 바꿔주었다. 

 

 

 

 


다른 풀이

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

using namespace std;

vector<string> solution(vector<string> players, vector<string> callings)
{
    vector<string> answer;

    map<string, int> m;
    for(int i=0; i<players.size(); i++)
        m[players[i]]=i;


    int s1, s2;
    string tmp;
    for(int i=0; i<callings.size(); i++)
    {
        s2=m[callings[i]]; s1=s2-1;
        m[players[s2]]--; m[players[s1]]++;
        //cout << s1 << " " << s2 << endl;
        tmp=players[s2];
        players[s2]=players[s1];
        players[s1]=tmp;
    }

    answer=players;

    return answer;
}

 

 


https://school.programmers.co.kr/learn/courses/30/lessons/178871?language=cpp 

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr