모서리 IT 블로그

문제 설명

스트리밍 사이트에서 장르 별로 가장 많이 재생된 노래를 두 개씩 모아 베스트 앨범을 출시하려 합니다. 노래는 고유 번호로 구분하며, 노래를 수록하는 기준은 다음과 같습니다.

1. 속한 노래가 많이 재생된 장르를 먼저 수록합니다.
2. 장르 내에서 많이 재생된 노래를 먼저 수록합니다.
3. 장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록합니다.

노래의 장르를 나타내는 문자열 배열 genres와 노래별 재생 횟수를 나타내는 정수 배열 plays가 주어질 때, 베스트 앨범에 들어갈 노래의 고유 번호를 순서대로 return 하도록 solution 함수를 완성하세요.

제한사항

• genres[i]는 고유번호가 i인 노래의 장르입니다.
• plays[i]는 고유번호가 i인 노래가 재생된 횟수입니다.
• genres와 plays의 길이는 같으며, 이는 1 이상 10,000 이하입니다.
• 장르 종류는 100개 미만입니다.
• 장르에 속한 곡이 하나라면, 하나의 곡만 선택합니다.
• 모든 장르는 재생된 횟수가 다릅니다.

입출력 예

내 문제 풀이

- genre와 plays 같은 index 값 = 같은 노래 이다.
1. 장르 별 전체 plays 계산한다.
2. 장르 별 전체 plays 기준으로 내림차순 정렬한다. (문제 조건 1)
3. 장르 하나마다 반복문 수행하며 해당 장르 노래의 index 값을 알아낸다. 이 index를 사용해서 노래 별 plays를 top_song 리스트에 저장한다.
4. 해당 장르의 노래가 2개 미만이라면, 해당 plays를저장한다.
5. 해당 장르의 노래가 여러 개라면, plays 기준으로 내림차순 정렬 후 앞 2개의 plays를 저장한다. 이 때 index()는 리스트 앞 부분부터 찾기 때문에 중복된 plays가 있는 경우, 첫 번째로 저장한 노래의 고유 번호를 중복해서 answer에 저장하지 않기 위해 임의로 -1 값으로 변경해준다.

Collections.defaultdict

객체마다의 카운트 갯수를 알고 싶을 때 collections.Counter 말고 Collections.defaultdict를 사용할 수 있다. key-value 형태의 dictionary 형태로 저장할 수 있다. 
또한 정렬을 할 때에는 sort()를 사용할 수 없고 대신 sorted()를 사용한다.

내 코드 - 파이썬

from collections import defaultdict

def solution(genres, plays):
    answer = []
    
    song_count = defaultdict(int)
    for idx in range(0, len(genres)):
        song_count[genres[idx]] += plays[idx]
    # -----------------> genre별 전체 play count 완료
    
    song_count = sorted(song_count.items(), key=lambda x:x[1], reverse=True)
    # defaultdict는 sort() 사용 불가능
    
    # 상위 장르 순서대로 반복
    for sc in song_count:
        # 현재 장르 index(노래)의 plays 저장
        top_song = []        
        for idx in range(0, len(genres)):
            if genres[idx] == sc[0]:
                top_song.append(plays[idx])
                
        # 노래가 한 개일 때
        if len(top_song) < 2:
            answer.append(plays.index(top_song[0]))
        
        # 노래가 여러 개일 때
        else:
            # plays 수를 내림차순 정렬 
            top_song.sort(reverse=True)
            answer.append(plays.index(top_song[0]))
            plays[plays.index(top_song[0])] = -1
            answer.append(plays.index(top_song[1]))

    return answer

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.
예를 들어 스파이가 가진 옷이 아래와 같고 오늘 스파이가 동그란 안경, 긴 코트, 파란색 티셔츠를 입었다면 다음날은 청바지를 추가로 입거나 동그란 안경 대신 검정 선글라스를 착용하거나 해야 합니다.

스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때 서로 다른 옷의 조합의 수를 return 하도록 solution 함수를 작성해주세요.

제한사항

• clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.
• 스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.
• 같은 이름을 가진 의상은 존재하지 않습니다.
• clothes의 모든 원소는 문자열로 이루어져 있습니다.
• 모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.
• 스파이는 하루에 최소 한 개의 의상은 입습니다.

입출력 예

내 문제 풀이

1. 의상 종류 별 의상 이름 갯수를 센다.
2. 전체 조합 계산 = (n+1) * (m+1) * ... 을 활용하여 의상 종류 별 의상 이름 갯수를 모두 곱한다.
3. 최소 한 개의 의상은 입어야 하므로 아무것도 입지 않을 경우를 제외하고 return. (2번 결과의 -1)

collections.Counter

동일한 값의 갯수를 count 해주며 dictionary 형태로 반환된다. 
Counter('abcaab')을 수행할 경우 Counter({'a':3, 'b':2, 'c':1})이 return 된다.

내 코드 - 파이썬

from collections import Counter

def solution(clothes):
    answer = 1
    

    counter = Counter(cloth[1] for cloth in clothes)
    for a, b in counter.items():
        answer *= (b+1)
    
    return answer-1

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

• 구조대 : 119
• 박준영 : 97 674 223
• 지영석 : 11 9552 4421

전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
• 각 전화번호의 길이는 1 이상 20 이하입니다.

입출력 예제

내 문제 풀이

1. phone_book을 정렬한다.(오름차순, 내림차순 상관 없음.)
2. phone_book 기준으로 반복문을 수행한다. 반복 횟수는 phone_book 길이 -1.
3. 반복문을 수행하며 현재 index의 전화번호가 바로 뒤 index의 전화번호에 포함되는지 확인한다.
4. 포함되는 된다면 바로 False를 return.
5. 반복문이 끝날 때까지 return이 되지 않으면 서로 접두사인 전화번호가 없다는 것이므로 True를 return.

내 코드 - 파이썬

def solution(phone_book):
    phone_book.sort()
    
    for idx in range(0, len(phone_book)-1):
        if phone_book[idx] in phone_book[idx+1]:
            return False
        
    return True

문제 설명

수많은 마라톤 선수들이 마라톤에 참여하였습니다. 단 한 명의 선수를 제외하고는 모든 선수가 마라톤을 완주하였습니다.
마라톤에 참여한 선수들의 이름이 담긴 배열 participant와 완주한 선수들의 이름이 담긴 배열 completion이 주어질 때, 완주하지 못한 선수의 이름을 return 하도록 solution 함수를 작성해주세요.

제한사항

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

입출력 예

내 문제 풀이

1. participant와 completion을 오름차순으로 정렬한다.(내림차순도 상관 없음)
2. participant 기준으로 반복문을 실행한다. 같은 index의 participant와 completion 선수 이름 비교. 
   이 때 반복문의 index는 len(participant)-1까지 수행한다. (문제에서 completion 길이= participant 길이 -1 로 주어짐.)
3. 아래 그림처럼 다른 경우가 발생하면 바로 해당 index의 participant 선수 이름 return.

4. 반복문이 끝나면 participant의 가장 마지막 선수가 완주하지 못한 것이므로 해당 선수 이름 return.

내 코드 - 파이썬

def solution(participant, completion):
    answer = ''
    participant.sort(reverse=True)
    completion.sort(reverse=True)
    
    for par in range(0, len(participant)-1):
        if participant[par] != completion[par]:
            return participant[par]
        
    return participant[-1]