[파이썬, Java] 완주하지 못한 선수

문제 설명

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

마라톤에 참여한 선수들의 이름이 담긴 배열 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"는 참여자 명단에는 두 명이 있지만, 완주자 명단에는 한 명밖에 없기 때문에 한명은 완주하지 못했습니다.

출처

 

 

👸 나의 풀이

def solution(participant, completion) :
    dic={}
    for p in participant :    
        dic[p]=dic.get(p, 0)+1
    for c in completion :
        dic[c]-=1
    return [k for k, v in dic.items() if v==1][0]

해시

위 풀이의 시간 복잡도는 O(N)

딕셔너리의 get 메소드는 get(key, 기본값)이다.

key에 값이 없으면 지정한 기본값 0을 넣는다.

key에 값이 있으면 값을 가져온다.

 

 

from collections import Counter

def solution(participant, completion) :
    ans=Counter(participant)-Counter(completion)
    return ''.join(ans.keys())

 

복습겸 다시 풀어봄

Counter는 문자열 비교할 때 사용하기 좋다.

 

# 통과
def solution(participant, completion):  # 전체 선수들, 완주한 선수들
    participant.sort()
    completion.sort()
    
    for i in range(len(completion)) :
        if participant[i] != completion[i] :
            return participant[i]   
    return participant[-1]        # 가장 끝 값

 

마라톤에 참가한 선수들 participant와 완주한 선수들 completion을 오름차순 정렬했다.

반복문을 돌려서 두 값이 다르면 participant[i]는 완주 못한 선수다.

끝까지 봐도 안 나오면 participant에서 마지막 선수가 답이 된다.

해쉬로 풀었다면 더 좋았을텐데

 

 

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        int len=participant.length;
        Arrays.sort(participant);
        Arrays.sort(completion);

        for(int i=0; i<len-1; i++){
            if(!participant[i].equals(completion[i])){
                return participant[i];
            }
        }
        
        return participant[len-1];
    }
}

 

participant와 completion을 정렬후 원소를 비교한다.

반복문 내에서 다른 원소가 나오면 그 값을 리턴한다.

끝까지 못찾으면 마지막 원소를 반환한다.

 

 

# 실패
def solution(participant, completion):  # 전체 선수들, 완주한 선수들
    dic={}
    person=set()
    
    # 전체 선수 등록
    for p in participant :
        if p in dic.keys() :	# 동명이인일 경우
            dic[p]+=1
            person.add(p)
        else:
            dic[p]=0 
            
    # 동명이인 참가자와 완주한 동명이인 수 비교
    for p in person :
        if dic[p]+1 != completion.count(p) :
            return p
    
    # 참가자 체크
    for c in completion : 
        dic[c]+=1
        
    return "".join([p for p in dic if dic[p] == 0 ])   # 완주하지 못한 선수   

 

첨에 제출하니 60점..ㅠ

시간초과가 났다. for문이 너무 많아서;;ㅎ

 

 

👱‍♂️ 다른 사람 풀이

 

import collections

def solution(participant, completion):  # 전체 선수들, 완주한 선수들
    answer=collections.Counter(participant)-collections.Counter(completion)
    return list(answer)[0]

 

collections내의 내장함수 Counter를 사용하면 딕셔너리가 만들어진다.

딕셔너리끼리 차를 구해서 answer에 담는다.

answer.keys()로 리턴을 할려고 하면 TypeError: Object of type dict_keys is not JSON serializable 에러가 뜬다. 

return ''.join(answer.keys())는 가능하다.

 

 

def solution(participant, completion):  # 전체 선수들, 완주한 선수들
    answer=''
    temp=0
    dic={}
    
    for part in participant :
        dic[hash(part)]=part
        temp+=int(hash(part))
    for com in completion :
        temp-=hash(com)
    answer=dic[temp]
    
    return answer

 

딕셔너리와 해시 함수를 이용한 풀이

key를 선수 이름 해시, 값을 선수이름으로 한다. 

hash()는 객체의 해시값을 반환한다. 해시값은 정수다.  

completion을 반복하면서 temp에서 hash값을 빼간다.

반복이 다 끝나면 완주 못한 선수가 남는다.

딕셔너리 dic에 키 temp를 넣고 answer을 리턴한다.

 

 

def solution(participant, completion):  # 전체 선수들, 완주한 선수들
    participant.sort()
    completion.sort()
    
    for p, c in zip(participant, completion) :
        if p != c :
            return p
    return participant[-1]

 

zip을 이용해서 반복문을 좀 더 효율적으로 사용했다.

zip(*iterables)은 입력 받은 iterables들에서 똑같은 인덱스에 있는 값들을 뽑아낸다.

 

 

// https://dreamhollic.tistory.com/entry/%EC%95%8C%EA%B3%A0%EB%A6%AC%EC%A6%98-%EB%AC%B8%EC%A0%9C-%ED%92%80%EC%9D%B45-%EC%99%84%EC%A3%BC%ED%95%98%EC%A7%80-%EB%AA%BB%ED%95%9C-%EC%84%A0%EC%88%98-JAVA
import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer="";
        int val=0;
        
        Map <String, Integer> hm=new HashMap<>();
        
        for(String part:participant){
            if(hm.get(part) == null){
                hm.put(part, 1);
            }else{
                val=hm.get(part)+1;
                hm.put(part, val);
            }
        }
        
        for(String comp:completion){
            val=hm.get(comp)-1;
            hm.put(comp, val);
        }
        
        for(String key:hm.keySet()){
            if(hm.get(key) == 1) answer=key;
        }
        return answer;
    }
}

 

해시 사용

participant의 요소를 key, 값은 key의 갯수다.

completion의 요소를 key로 값을 1씩 감소시킨다.

key의 값이 1인 key가 답이다.

 

 

 

문제 출처 💁‍♀️ 프로그래머스