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

프로그래머스

 

📄 문제

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

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

 

📝 풀이

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

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

	// 두개의 map을 사용하여 swap
	map<int, string> m1;
	map<string, int> m2;

	// map에 모든 정보 삽입
	for (int i = 0; i < players.size(); i++)
	{
		m1[i] = players[i];
		m2[players[i]] = i;
	}

	// calling 만큼 반복
	for (int i = 0; i < callings.size(); i++)
	{
		// 현재 이름을 가리키는 인덱스 번호를 빠르게 가져오기
		int currentIdx = m2[callings[i]];

		// callings[1]보다 한칸 앞에 있는 대상을 가져오기
		string change = m1[currentIdx - 1];

		// m1을 앞뒤로 변경
		m1[currentIdx - 1] = callings[i];
		m1[currentIdx] = change;

		// m2를 앞뒤로 변경
		m2[callings[i]] = currentIdx - 1;
		m2[change] = currentIdx;
	}

	for (pair<int, string> p : m1)
		answer.push_back(p.second);
    
	return answer;
}

• vector 자료구조를 swap하여 사용하려고 했었으나, 문제 제한사항의 범위가 매우 크기에 타임리밋이 걸릴 것 같아 새로운 방법인 두개의 map을 이용하여 인덱스 번호를 통해 빠르게 접근하여 swap하는 방법으로 해결하였습니다.