IT-garden

문제 설명

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

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

제한조건

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

입출력 예

입출력 예 설명

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

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

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

나의 풀이

def solution(participant, completion):
    participant.sort()
    completion.sort()
    
    answer = participant[-1]
    for i in range(len(completion)):
        if participant[i] != completion[i]:
            answer = participant[i]
            break
            
    return answer

1. 참여한 선수들의 배열과 완주한 선수들의 배열을 정렬합니다.
2. 두 배열을 비교하여 값이 다른 경우 참여한 선수 배열의 값을 출력해줍니다.

다른 사람의 풀이

import collections

def solution(participant, completion):
    answer = collections.Counter(participant) - collections.Counter(completion)
    return list(answer.keys())[0]

Counter로 리스트의 원소들을 각각 세서 원소들을 key로, 등장하는 개수를 value로 하여 dictionary를 만들어 빼주었습니다. 그리고 마라톤에 참여했지만 완주하지 못한 선수를 key값으로 출력해주었습니다.

프로그래머스 '완주하지 못한 선수' : programmers.co.kr/learn/courses/30/lessons/42576?language=python3

Counter는 collections의 모듈인 Counter 클래스를 사용합니다.

Counter는 항목의 개수를 세어줄 때 유용하게 사용할 수 있습니다.

from collections import Counter

a=[1,2,3,4,5,1,3,5,1,5,5]
b='abcdeaceabaeaa'

# 출력>>Counter({5: 4, 1: 3, 3: 2, 2: 1, 4: 1})
print(Counter(a))

# 출력>>Counter({'a': 6, 'e': 3, 'b': 2, 'c': 2, 'd': 1})
print(Counter(b))

Counter는 딕셔너리 형태로 Key 값으로 요소의 이름, Value 값으로 요소들의 횟수를 출력해줍니다.

from collections import Counter

b='abcdeaceabaeaa'
c='eabdeabc'

# 출력>>Counter({'a': 6, 'e': 3, 'b': 2, 'c': 2, 'd': 1})
print(Counter(b))

# 출력>>Counter({'e': 2, 'a': 2, 'b': 2, 'd': 1, 'c': 1})
print(Counter(c))

# 출력>>Counter({'a': 8, 'e': 5, 'b': 4, 'c': 3, 'd': 2})
print(Counter(b) + Counter(c))

# 출력>>Counter({'a': 4, 'c': 1, 'e': 1})
print(Counter(b) - Counter(c))

# 출력>>Counter()
print(Counter(c) - Counter(b))

# 출력>>[('a', 6), ('e', 3)]
print(Counter(b).most_common(2))

Counter를 이용해 계산도 가능합니다.

c에서 b를 빼면 c가 전체적으로 더 작기 때문에 빈 값으로 나옵니다.

most_common('상위요소 수')를 통해 상위요소를 출력할 수 있습니다.
상위요소의 수를 쓰지 않으면 전체요소를 출력해줍니다.

count와 Counter는 모두 항목의 개수를 세어줄 때 많이 사용합니다.

count의 시간 복잡도는 O(N)입니다.
여기서 각각의 리스트의 개수를 확인하기 위해서는 for문을 통해 count를 n번하기 때문에 시간 복잡도는 O(N2)입니다.

Counter 클래스를 이용해 Counter를 n번 하면 시간 복잡도는 O(N)입니다.
딕셔너리에서 원소를 접근할 때의 시간 복잡도는 O(1)이기 때문입니다.

from collections import Counter

a=[0, 1, 2, 3, 4, 0, 2, 4, 0, 2, 1, 1, 1, 1, 1, 2, 4]

# count 사용
for i in range(5):
    # 출력>>3 6 4 1 3
    print(a.count(i),end=' ')

# Counter 사용
counter = Counter(a)
for i in range(5):
    # 출력>>3 6 4 1 3
    print(counter[i],end=' ')

count를 여러번 사용하는 경우는 Counter 클래스가 더욱 유용합니다.