본문 바로가기

알고리즘 공부

마라톤 선수 -프로그래머스 hash 문제-

728x90
반응형

 

문제 설명

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

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

 

첫 번째 풀이(set을 이용한 방법)


문제를 처음 봤을 때 set을 이용하여 두 집합을 빼서 남는 선수를 return 해주어 답을 찾으려고 하여 코드를 작성하였다

def solution(participant, completion):
    answer = ''
    a=[]
    a = list(set(participant)-set(completion))
    answer = a[-1]
    return answer

 

set으로 하였을 때 문제가 발생하였는데 집합을 만들면 동명이인이 있을 경우 한 명으로 판단한다

즉 예제 3번 같은 경우에 mislav가 두명있는데 set을 사용하면 {mislav, stanko, ana}로 되며

전부 완주했다고 나오기 때문에 다른 방법을 찾아야했다.

 

 

두 번째 풀이(sort를 이용한 풀이)


#두 리스트를 sort를 이용하여 순서대로 만든뒤 앞에서 부터 비교 하여 만약 다를 경우 
#완주하지 못했다고 간주해서 answer에 저장하고 break로 빠져나온다
def solution(participant, completion):
    answer = ''
    participant.sort()
    completion.sort()
    for i in range(len(completion)):
        if participant[i] != completion[i]:
            answer = participant[i]
            break
#for문을 다 돌았는데 answer가 정해지지 않은 경우는 participant마지막에 있는 선수가 완주하지 못한것이므로 
#마지막을answer에 저장한다
    if answer == '':
        answer = participant[-1]
        
    return answer

 

sort를 이용하여 particparticipant, completion이 순서대로 정렬될 수 있게 만든다

for문을 이용하여 completion 수만큼 검사를 한다 이때 만약 다르다면 완주하지 못한 선수이기 때문에

answer에 저장하고 break

끝까지 돌았는데 틀린 게 없다면 participant리스트에 마지막 선수가 완주하지 못한 것이므로

answer에 마지막 선수를 저장한다

 

 

java로 풀이


알고리즘 문제 중 python이 아니라 java로만 풀어야 하는 경우가 있어서 java로도 작성해보았다

import java.util.*;
class Solution {
    public String solution(String[] participant, String[] completion) {
        String answer = "";
        Arrays.sort(participant);
        Arrays.sort(completion);
        int i;
        for(i = 0; i < completion.length; i++){
            if(!participant[i].equals(completion[i])){
                answer = participant[i];
                    break;
            }
        }
        answer = participant[i];
        return answer;
    }
}

 

풀이 방식은 python과 같은 알고리즘으로 풀었다

 

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges

728x90
반응형