본문 바로가기
개인 공부/코딩테스트

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

by seowit 2020. 8. 9.

문제

 

문제 설명

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

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

제한사항

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

입출력 예

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

입출력 예 설명

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

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

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

 


[ C++ ]

 

풀이 1 _ sort

 

participant 와 completion 벡터를 모두 정렬한다. 

더보기

sort ?

추가 헤더파일 : #include <algorithm>

사용법 : sort(start, end, 함수) - 함수에는 사용자 지정 함수, greater<type>(), less<type>() 3가지가 올 수 있다.

사용예시 : sort(v.begin(), v.end(), greater<int>()); - v는 vector

sort ?

추가 헤더파일 : #include <algorithm>

사용법 : sort(start, end, 함수) - 함수에는 사용자 지정 함수, greater<type>(), less<type>() 3가지가 올 수 있다.

사용예시 : sort(v.begin(), v.end(), greater<int>()); - v는 vector

정렬 후, participant가 [a, b, c, d, e] 이고, completion이 [a, b, c, e] 라고 해보자.

두 벡터를 앞에서부터 비교를 하면 첫 번째 원소는 a로 같고, 두 번째 원소도 b로 같고, 세 번째 원소도 c로 같다. 네 번째 원소가 participant는 d이고 completion은 e로 달라진다. 

달라지는 순간의 participant 원소(d)는 completion에 없는 원소가 된다. 

따라서 그 값을 answer로 반환하면 된다.

 

소스코드

 

#include <string> // string 사용을 위한 헤더
#include <vector> // vector 사용을 위한 헤더
#include <algorithm> // sort를 사용하기 위한 헤더

using namespace std;

string solution(vector<string> participant, vector<string> completion) {
    string answer = "";
	
    // participant와 completion 벡터를 정렬한다.
    sort(participant.begin(), participant.end());    
    sort(completion.begin(), completion.end());    
	
    // participant와 completion의 원소 하나씩 비교한다.
    for (int i = 0; i < participant.size(); i++)
    {	// 다른 원소를 반환한다.
        if (participant[i] != completion[i])
        {
            return participant[i];
        }
    }
    return answer;
}

 


 

풀이 2 _ hash  (다른사람 풀이)

 

completion의 원소를 unordered_map에 저장한다.

participant의 원소 중에서 unordered_map에 저장되지 않은 원소를 찾는다.

더보기

unordered_map ?

hash table을 이용한 배열로, key와 value가 있다. python의 딕셔너리와 같다.

문자열을 인덱스로 사용할 수 있다는 장점이 있다. 그러나 데이터가 많을 때는 map을 사용하는 것이 더 효율적이다.

추가 헤더파일 : #include <unordered_map>

사용법 :

  • 선언 : unordered_map<key type, value type> m; (m은 변수명)
  • 추가 : m[key] = value, m.insert(make_pair(key, value))
  • 삭제 : m.erase(key), m.clear()
  • 조회 : m[key] (value 여러개면 맨 앞 원소 반환, 없으면 0 반환), m.find(key) (해당 key값의 iterator 반화) 
  • 그 외 : m.count(key), m.empty(), m.begin(), m.end(), m.size()

unordered_map ?

hash table을 이용한 배열로, key와 value가 있다. python의 딕셔너리와 같다.

문자열을 인덱스로 사용할 수 있다는 장점이 있다. 그러나 데이터가 많을 때는 map을 사용하는 것이 더 효율적이다.

추가 헤더파일 : #include <unordered_map>

사용법 :

  • 선언 : unordered_map<key type, value type> m; (m은 변수명)
  • 추가 : m[key] = value, m.insert(make_pair(key, value))
  • 삭제 : m.erase(key), m.clear()
  • 조회 : m[key] (value 여러개면 맨 앞 원소 반환, 없으면 0 반환), m.find(key) (해당 key값의 iterator 반화) 
  • 그 외 : m.count(key), m.empty(), m.begin(), m.end(), m.size()

 

소스코드

 

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

using namespace std;

string solution(vector<string> participant, vector<string> completion) {
    string answer = "";
    unordered_map<string, int> strMap;
    for(auto elem : completion)
    {
        if(strMap.end() == strMap.find(elem))
            strMap.insert(make_pair(elem, 1));
        else
            strMap[elem]++;
    }

    for(auto elem : participant)
    {
        if(strMap.end() == strMap.find(elem))
        {
            answer = elem;
            break;
        }
        else
        {
            strMap[elem]--;
            if(strMap[elem] < 0)
            {
                answer = elem;
                break;
            }
        }
    }
    return answer;
}

 

<tip>

 

if(strMap.end() == strMap.find(elem)) 

strMap.end()는 마지막값 다음의 메모리 주소(비어있음)를 반환한다.

만약 strMap에 elem이 없다면 strMap.find(elem)도 마지막값 다음의 메모리 주소를 반환하므로

" strMap.end() == strMap.find(elem) " 는  strMap에 찾는 값(elem)이 없음을 의미한다.

 


[ python ]

 

소스코드 - sort 이용

 

def solution(participant, completion):
    sorted_participant = sorted(participant)
    sorted_completion = sorted(completion)
    answer = sorted_participant[-1]
    for x, y in zip(sorted_participant, sorted_completion):
        if(x != y) :
            answer = x
            break
    return answer

 

댓글