본문 바로가기

문제풀이

[프로그래머스] 완주하지 못한 선수 C++

[문제 설명]
수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.

마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

[제한사항]
마라톤 경기에 참여한 선수의 수는 1명 이상 100,000명 이하입니다.
completion의 길이는 participant의 길이보다 1 작습니다.
참가자의 이름은 1개 이상 20개 이하의 알파벳 소문자로 이루어져 있습니다.
참가자 중에는 동명이인이 있을 수 있습니다.

 

[문제 풀이]

#include <string>
#include <vector>
#include <unordered_map>

using namespace std;

string solution(vector<string> participant, vector<string> completion) {
    string answer = "";
    //해시맵 생성
   unordered_map<string, int> map; 
    
    for(auto member : participant){
        //member를 찾지 못하면 map에 추가
        if(map.end() == map.find(member))
            map.insert(make_pair(member, 1));
        //member 찾으면 1씩 더하기
        else map[member]++;
    }
    //완주한 member 빼기
    for(auto member : completion) map[member] --;
    
    for(auto member : participant){
        //완주하지 못한 member 반환
        if(map[member] > 0) {
            answer = member;
            break;
        }
    }
    return answer;
}

 

[학습 사항 / 개선 사항]

if(map.end() == map.find(member)) 에서 왜 map.end()가 들어가는지 이해하지 못해서 찾아봤다.

end 이터레이터가 가리키는 위치는 데이터 값이 없기 때문에 find함수가 특정 값을 찾지 못하면 반환하는 값과 같아서였다.

 

해시 카테고리에 있는 문제여서 해시로 풀 생각만 했는데, 다른 사람의 풀이를 보니 단순 정렬해서 두 배열을 비교해도 풀리는 문제였다. 이 풀이방법이 더 간단해서 마음에 든다.