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

문제

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

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

 

제한사항

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

 

입출력 예제

participant	completion	return
["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"는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

풀이

문제를 보면 "단 한명의 선수를 제외하고" 모든 선수가 완주한다는 문구가 있다.

다시 말해 [출력 수]는 [입력 수 - 1] 이라는 의미로 이해할 수 있다.

예를 들어 2, 2, 3, 4, 3 선수, 총 5명이 출전하면, 완주하는 선수는 4명이고,

완주한 선수가 3, 2, 3, 4 라고 가정할 때, 2번 선수 중 하나는 완주하지 못하였다는 의미이다.

 

이를 순서대로 나타내면 아래와 같다 (정렬)

입력 : 2, 2, 3, 3, 4

출력 : 2, 3, 3, 4

출력과 입력은 1개를 제외하고, 모든 값을 동일하게 갖기 때문에,

처음 인덱스부터 1쌍씩 비교를 하면서 차이점이 발생하는 첫 부분, 즉 인덱스 1번의 (2, 3)에서 완주하지 못한 선수를 찾을 수 있다.

#include <algorithm>  //sort를 위해서

string solution(vector<string> participant, vector<string> completion)
{
  string answer = "";

  sort(participant.begin(), participant.end());
  sort(completion.begin(), completion.end());

  unsigned int index;
  for (index = 0; index < completion.size(); index++)
  {
    if (participant[index] != completion[index])
    {
      answer = participant[index];
      break;
    }
  }

  /*
  차이점이 발생한 인덱스를 찾게될 경우 break로  빠져나와,
    index는 completion.size()가 될 수 없다.

  위에서 값을 찾지 못한 경우
    ex. 입력 : 1, 2, 3, 4, 5
    ex. 출력 : 1, 2, 3, 4
    */
  if (index == completion.size())
  {
    answer = participant[index];
  }

  return answer;
}

 

다른 풀이

해당 문제의 태그는 "해시" 이고, 해시는 Key와 Value로 구분되어 있다는 말에 따라,

C++ STL에서 이와 유사한 Map을 통해 아래 방식을 접근해보면 다음과 같다.

(굳이 맵을 이용하지 않고, 배열을 활용해도 시간 초과는 발생하지 않는다.)

#include <algorithm>
#include <map>

string solution(vector<string> participant, vector<string> completion)
{
  string answer = "";

  map<string, int> mapPair;  // Map : Key + Value

  vector<string>::iterator iter;

  // Map에 참여자를 등록한다. 중복된 참여자가 나올 경우, value값을 +1 시킨다.
  for (iter = participant.begin(); iter != participant.end(); iter++)
  {
    if (mapPair.find(*iter) == mapPair.end())
    {
      mapPair.insert(make_pair(*iter, 1));
    }
    else
    {
      mapPair[*iter]++;
    }
  }

  // 등록한 Map에서 완주자를 제외시킨다.
  for (iter = completion.begin(); iter != completion.end(); iter++)
  {
    mapPair[*iter]--;
  }

  // 결과적으로 value값이 1인 인덱스의 key값이 정답이다.
  map<string, int>::iterator mapIter;

  for (mapIter = mapPair.begin(); mapIter != mapPair.end(); mapIter++)
  {
    if (mapIter->second != 0)
    {
      answer = mapIter->first;
      break;
    }
  }

  return answer;
}

추가적으로 위 코드에서 undordered_map을 활용할 경우, 약간의 검사 시간 차이를 극복할 수 있으나,

해당 맵 라이브러리도, key 값이 string으로 넘어갈 경우 비약적인 속도 증가는 얻기 힘들다.

 

아래 코드는 학습용으로 만들어본 Python 코드이다.

def solution(participant, completion):
    answer = ''

    dicPart = {}

    for i in range(len(participant)) : 

        if participant[i] in dicPart.keys() :
            dicPart[participant[i]] += 1
        else :
            dicPart[participant[i]] = 1

    for i in range(len(completion)) :

        dicPart[completion[i]] -= 1

    for i in dicPart :

        if dicPart[i] == 1 :

            answer = i

    return answer

 

출처

문제 : https://programmers.co.kr/learn/courses/30/lessons/42576