프로그래머스 해시 완주하지 못한 선수

문제 (링크)

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

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

제한사항

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

입출력 예

participantcompletionreturn

[leo,kiki,eden]	[eden,kiki]	leo
[marina,josipa,nikola,vinko,filipa]	[josipa,filipa,marina,nikola]	vinko
[mislav,stanko,mislav,ana]	[stanko,ana,mislav]	mislav

입출력 예 설명

예제 #1
leo는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #2
vinko는 참여자 명단에는 있지만, 완주자 명단에는 없기 때문에 완주하지 못했습니다.

예제 #3
mislav는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

 

 

 

처음엔 해시니까 맵 써서 하면 되겠지, 하고 map을 사용했다.

아직 c++에 적응이 안돼서 오류 계속 나고 한참 헤멤...

map을 쓰기로 하고 string-bool pair의 자료구조를 쓰기로 했다.

코드 작성하고 제출해보니 문법때문에 계속 에러나서 진짜 울뻔했음

 

알고보니 멍청하게 콜론을 빼먹은 거였다..! 타스 프리티어에서 콜론 안쓰는게 습관이 돼버려서...

 

아무튼 드디어 에러는 나지 않게 수정한 코드다

 

string solution(vector<string> participant, vector<string> completion) {
    map<string, bool> isCompleted;
    
    // 초기화
    for(string p: participant) {
        isCompleted[p] = false;
    }
    
    for(string c: completion) {
        isCompleted[c] = true;    
    }
    
    for(auto name: participant) {
        if (!isCompleted[name]) {
        return name;
        }
    }
}

 

 

당근 통과하겠지 생각하고 넣었는데 아뿔사 세번째 테스트 케이스를 통과를 못했다.

[mislav, stanko, mislav, ana]

[stanko, ana, mislav]

mislav

동명이인이 가능하기 때문에 participants['mislav']가 true되어서 완주한 것으로 된다.

 

따라서 bool이 아니라 int를 사용하여 동명이인이 있는 케이스를 다룰 수 있게 수정하였다.

string solution(vector<string> participant, vector<string> completion) {
    map<string, int> isCompleted;
    
    for(string p: participant) {
        ++isCompleted[p];
    }
    
    for(string c: completion) {
        --isCompleted[c];      
    }
    
    for(auto item: isCompleted) {
        if (item.second > 0 ) {
            return item.first;
        }
    }
}

 

세 개 테스트 케이스 통과하길래 정답 제출했다.

굿!

 

 

개선 아이디어

1. map이 아니라 unordered_map 이라는 자료구조가 있던데, 이건 굳이 정렬이 필요 없는 문제이므로 이 자료구조를 사용하면 효율성이 증가할 수 있을 듯하다.
2. 지금 내 코드는 O(n)의 시간 복잡도를 가졌다. 정확히는 3 * O(n). 하나하나 하지 않고 그냥 한번에 싹 비교하면 2 * O(n)으로 줄일 수 있을 것 같다. 2/3배면 꽤 큰 차이니까 유의미할듯.

 

다른 사람 코드 분석

string solution(vector<string> participant, vector<string> completion) {
    string answer = "";
    
    sort(participant.begin(), participant.end());
    sort(completion.begin(), completion.end());
    
    for(int i=0;i<completion.size();i++)
    {
        if(participant[i] != completion[i])
            return participant[i];
    }
    
    return participant[participant.size() - 1];
}

참가자와 완주자를 정렬하고 비교한 방법이다.

참신하긴 하지만 정렬은 O(n*log n)의 시간 복잡도를 가지니 시간 복잡도 측면에서 좋은 방법은 아닐 듯 하다.

물론 n이 작아서 (n*log n) < (3*n) 일 때는 그게 좀 더 빠르겠지만... 게다가 문제 취지에도 안맞다.

 

오랜만에 알고리즘 푸니까 문법도 그렇고 시행착오가 꽤 있었지만 재밌었다. 문제가 쉬워서 인 듯.. 계속 쉬운 문제만 풀고 싶다ㅜㅜ

다음 시간까지 화이팅~! 이제 일하러 가야지ㅎㅎ