프로그래머스 코딩테스트 고득점 Kit (2) 해시

해시

• 설명 : Key-value쌍으로 데이터를 빠르게 찾아보세요.
• 출제 빈도 : 높음
• 평균 점수 : 보통

---

폰켓몬 (Level 1, 29813명 완료)

문제 설명

당신은 폰켓몬을 잡기 위한 오랜 여행 끝에, 홍 박사님의 연구실에 도착했습니다. 홍 박사님은 당신에게 자신의 연구실에 있는 총 N 마리의 폰켓몬 중에서 N/2마리를 가져가도 좋다고 했습니다.
홍 박사님 연구실의 폰켓몬은 종류에 따라 번호를 붙여 구분합니다. 따라서 같은 종류의 폰켓몬은 같은 번호를 가지고 있습니다. 예를 들어 연구실에 총 4마리의 폰켓몬이 있고, 각 폰켓몬의 종류 번호가 [3번, 1번, 2번, 3번]이라면 이는 3번 폰켓몬 두 마리, 1번 폰켓몬 한 마리, 2번 폰켓몬 한 마리가 있음을 나타냅니다. 이때, 4마리의 폰켓몬 중 2마리를 고르는 방법은 다음과 같이 6가지가 있습니다.

1. 첫 번째(3번), 두 번째(1번) 폰켓몬을 선택
2. 첫 번째(3번), 세 번째(2번) 폰켓몬을 선택
3. 첫 번째(3번), 네 번째(3번) 폰켓몬을 선택
4. 두 번째(1번), 세 번째(2번) 폰켓몬을 선택
5. 두 번째(1번), 네 번째(3번) 폰켓몬을 선택
6. 세 번째(2번), 네 번째(3번) 폰켓몬을 선택

이때, 첫 번째(3번) 폰켓몬과 네 번째(3번) 폰켓몬을 선택하는 방법은 한 종류(3번 폰켓몬 두 마리)의 폰켓몬만 가질 수 있지만, 다른 방법들은 모두 두 종류의 폰켓몬을 가질 수 있습니다. 따라서 위 예시에서 가질 수 있는 폰켓몬 종류 수의 최댓값은 2가 됩니다.
당신은 최대한 다양한 종류의 폰켓몬을 가지길 원하기 때문에, 최대한 많은 종류의 폰켓몬을 포함해서 N/2마리를 선택하려 합니다. N마리 폰켓몬의 종류 번호가 담긴 배열 nums가 매개변수로 주어질 때, N/2마리의 폰켓몬을 선택하는 방법 중, 가장 많은 종류의 폰켓몬을 선택하는 방법을 찾아, 그때의 폰켓몬 종류 번호의 개수를 return 하도록 solution 함수를 완성해주세요.

제한사항

• nums는 폰켓몬의 종류 번호가 담긴 1차원 배열입니다.
• nums의 길이(N)는 1 이상 10,000 이하의 자연수이며, 항상 짝수로 주어집니다.
• 폰켓몬의 종류 번호는 1 이상 200,000 이하의 자연수로 나타냅니다.
• 가장 많은 종류의 폰켓몬을 선택하는 방법이 여러 가지인 경우에도, 선택할 수 있는 폰켓몬 종류 개수의 최댓값 하나만 return 하면 됩니다.

입출력 예

nums	result
[3,1,2,3]	2
[3,3,3,2,2,4]	3
[3,3,3,2,2,2]	2

입출력 예 설명

입출력 예 #1
문제의 예시와 같습니다.

입출력 예 #2
6마리의 폰켓몬이 있으므로, 3마리의 폰켓몬을 골라야 합니다.
가장 많은 종류의 폰켓몬을 고르기 위해서는 3번 폰켓몬 한 마리, 2번 폰켓몬 한 마리, 4번 폰켓몬 한 마리를 고르면 되며, 따라서 3을 return 합니다.

입출력 예 #3
6마리의 폰켓몬이 있으므로, 3마리의 폰켓몬을 골라야 합니다.
가장 많은 종류의 폰켓몬을 고르기 위해서는 3번 폰켓몬 한 마리와 2번 폰켓몬 두 마리를 고르거나, 혹은 3번 폰켓몬 두 마리와 2번 폰켓몬 한 마리를 고르면 됩니다. 따라서 최대 고를 수 있는 폰켓몬 종류의 수는 2입니다.

풀이

나의 풀이...

def solution(nums):
    answer = len(nums) // 2
    nums.sort() #불필요
    nums = list(set(nums)) #여기서 set을 하면 sort할 필요가 없음.
    if answer > len(nums): #min() 함수 사용해서 한줄로 바꿀 수 있음.
        answer = len(nums)
    return answer

고수의 지리는 풀이...

def solution(ls):
    return min(len(ls)/2, len(set(ls)))

---

완주하지 못한 선수 (Level 1, 76150명 완료)

문제 설명

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

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

풀이

나의 풀이, 최근에 알게된 Counter로 다시 풀어보았다. 쉽네? ㅡ.ㅡ;

from collections import Counter

def solution(participant, completion):
    participant_counter = Counter(participant)
    completion_counter = Counter(completion)
    not_completion = participant_counter - completion_counter
    return list(not_completion)[0]

고수의 풀이 보니... 1줄 변태들도 있네;

from collections import Counter

def solution(participant, completion):
    return list(Counter(participant) - Counter(completion))[0]

---

전화번호 목록 (Level 2, 45861명 완료)

문제 설명

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

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

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

제한 사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
  • 각 전화번호의 길이는 1 이상 20 이하입니다.
  • 같은 전화번호가 중복해서 들어있지 않습니다.

입출력 예제

phone_book	return
["119", "97674223", "1195524421"]	false
["123","456","789"]	true
["12","123","1235","567","88"]	false

입출력 예 설명

입출력 예 #1
앞에서 설명한 예와 같습니다.

입출력 예 #2
한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3
첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

알림

2021년 3월 4일, 테스트 케이스가 변경되었습니다. 이로 인해 이전에 통과하던 코드가 더 이상 통과하지 않을 수 있습니다.

풀이

C++로는 이렇게 풀었었는데...

#include <string>
#include <vector>
#include <iostream>
#include <set>

using namespace std;

bool solution(vector<string> phone_book) {
    set<string> tmp;
    
    for (auto& number : phone_book)
        tmp.insert(number);
    
    for (auto it1 = tmp.begin(); it1 != tmp.end(); ++it1)
    {
        for (auto it2 = it1; it2 != tmp.end(); ++it2)
        {
            if(it1!=it2)
                if (it2->find(*it1) == 0)
                    return false;
                else
                    break;
        }
    }
    return true;
}

파이썬으로는 이렇게 바꾸면 깔끔하다.

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

고수의 정석 풀이 ㅡ.ㅡ;;;;

def solution(phone_book):
    answer = True
    hash_map = {}
    for phone_number in phone_book:
        hash_map[phone_number] = 1
    for phone_number in phone_book:
        temp = ""
        for number in phone_number:
            temp += number
            if temp in hash_map and temp != phone_number:
                answer = False
    return answer

O(N) 풀이. zip으로 하면 O(N)이라고 한다.

def solution(phoneBook):
    phoneBook = sorted(phoneBook)

    for p1, p2 in zip(phoneBook, phoneBook[1:]):
        if p2.startswith(p1):
            return False
    return True

---

위장 (Level 2, 41255명 완료)

문제 설명

스파이들은 매일 다른 옷을 조합하여 입어 자신을 위장합니다.

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

종류	이름
얼굴	동그란 안경, 검정 선글라스
상의	파란색 티셔츠
하의	청바지
겉옷	긴 코트

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

제한사항

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

입출력 예

clothes	return
[["yellow_hat", "headgear"], ["blue_sunglasses", "eyewear"], ["green_turban", "headgear"]]	5
[["crow_mask", "face"], ["blue_sunglasses", "face"], ["smoky_makeup", "face"]]	3

입출력 예 설명

예제 #1
headgear에 해당하는 의상이 yellow_hat, green_turban이고 eyewear에 해당하는 의상이 blue_sunglasses이므로 아래와 같이 5개의 조합이 가능합니다.

1. yellow_hat  2. blue_sunglasses  3. green_turban  4. yellow_hat + blue_sunglasses  5. green_turban + blue_sunglasses

예제 #2
face에 해당하는 의상이 crow_mask, blue_sunglasses, smoky_makeup이므로 아래와 같이 3개의 조합이 가능합니다.

1. crow_mask 2. blue_sunglasses 3. smoky_makeup

풀이

C++로 풀 때는 정석으로 풀었지만 Python으로 하니까 야매로 하게 되는 느낌...

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>

using namespace std;

int solution(vector<vector<string>> clothes) {
    int answer = 1;
    
    map<string,int> hashmap;
    map<string,int>::iterator iter;
    
    for(int i=0;i<clothes.size();i++)
    {
        if ( hashmap.find(clothes[i][1]) == hashmap.end() ) {
            hashmap[clothes[i][1]] = 1;
        } else {
            hashmap[clothes[i][1]]++;
        }
    }

    for(iter=hashmap.begin();iter!=hashmap.end();iter++) {
        cout << iter->first << " " << iter->second << endl;
        answer *= (iter->second+1);
    }    
    
    return answer - 1;
}

그대로 옮겨보면... 와 소스코드 길이 대박!

def solution(clothes):
    answer = 1
    hashmap = {}
    
    for i in clothes:
        if i[1] in hashmap:
            hashmap[i[1]] += 1
        else:
            hashmap[i[1]] = 1
    
    for i in hashmap.values():
        answer *= (i + 1)
         
    return answer - 1

---

베스트앨범 (Level 3, 25659명 완료)

문제 설명

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

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

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

제한사항

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

입출력 예

genres	plays	return
["classic", "pop", "classic", "classic", "pop"]	[500, 600, 150, 800, 2500]	[4, 1, 3, 0]

입출력 예 설명

classic 장르는 1,450회 재생되었으며, classic 노래는 다음과 같습니다.

• 고유 번호 3: 800회 재생
• 고유 번호 0: 500회 재생
• 고유 번호 2: 150회 재생

pop 장르는 3,100회 재생되었으며, pop 노래는 다음과 같습니다.

• 고유 번호 4: 2,500회 재생
• 고유 번호 1: 600회 재생

따라서 pop 장르의 [4, 1]번 노래를 먼저, classic 장르의 [3, 0]번 노래를 그다음에 수록합니다.

• 장르 별로 가장 많이 재생된 노래를 최대 두 개까지 모아 베스트 앨범을 출시하므로 2번 노래는 수록되지 않습니다.

※ 공지 - 2019년 2월 28일 테스트케이스가 추가되었습니다.

풀이

나의 C++ 풀이...

#include <string>
#include <vector>
#include <iostream>
#include <algorithm>
#include <map>
#include <tuple>

using namespace std;

#define TUP tuple<int,int,int,int,int>

vector<int> solution(vector<string> genres, vector<int> plays) {
    vector<int> answer;
    
    map<string, TUP> genres_map;
    vector<pair<int, string>> v;
    
    for(int i=0;i<genres.size();i++)
    {
        if(genres_map.find(genres[i]) == genres_map.end() )
        {
            genres_map[genres[i]] = make_tuple(plays[i],plays[i],i,-1,-1);
        }
        else
        {
            get<0>(genres_map[genres[i]]) += plays[i];
            
            if(plays[i] > get<1>(genres_map[genres[i]]) && plays[i] > get<3>(genres_map[genres[i]]))
            {
                if(get<1>(genres_map[genres[i]]) < get<3>(genres_map[genres[i]]))
                {
                    get<1>(genres_map[genres[i]]) = plays[i];    
                    get<2>(genres_map[genres[i]]) = i;    
                }
                else
                {
                    get<3>(genres_map[genres[i]]) = plays[i];    
                    get<4>(genres_map[genres[i]]) = i;    
                }
            }
            else
            {
                if(plays[i] > get<1>(genres_map[genres[i]]))
                {
                    get<1>(genres_map[genres[i]]) = plays[i];    
                    get<2>(genres_map[genres[i]]) = i;    
                }
                
                if(plays[i] > get<3>(genres_map[genres[i]]))
                {
                    get<3>(genres_map[genres[i]]) = plays[i];    
                    get<4>(genres_map[genres[i]]) = i;    
                }
            }
        }
    }
    for(map<string, TUP>::iterator itr = genres_map.begin(); itr != genres_map.end(); ++itr)
    {
        v.push_back(make_pair(get<0>(itr->second),itr->first));
    }
    
    sort(v.begin(), v.end(), greater<>());
    
    for(vector<pair<int, string>>::iterator iter = v.begin(); iter != v.end(); iter++)
    {   
        if(get<1>(genres_map[iter->second]) < get<3>(genres_map[iter->second]))
        {
            answer.push_back( get<4>(genres_map[iter->second]) );
            answer.push_back( get<2>(genres_map[iter->second]) );
        }
        else if(get<1>(genres_map[iter->second]) > get<3>(genres_map[iter->second]))
        {
            answer.push_back( get<2>(genres_map[iter->second]) );
            if(get<3>(genres_map[iter->second]) != -1)
                answer.push_back( get<4>(genres_map[iter->second]) );
        }
        else
        {
            if(get<2>(genres_map[iter->second]) < get<4>(genres_map[iter->second]))
            {
                answer.push_back( get<2>(genres_map[iter->second]) );
                answer.push_back( get<4>(genres_map[iter->second]) );   
            }
            else
            {
                answer.push_back( get<4>(genres_map[iter->second]) );
                answer.push_back( get<2>(genres_map[iter->second]) );   
            }
        }
    }
    
    return answer;
}

저걸 파이썬으로 옮긴... 나의 풀이

from collections import Counter

def solution(genres, plays):
    answer = []
    
    genres_total_play = {} # 장르 토탈 플레이시간 = 인기 장르 순위
    genres_song_count = {} # 장르별 곡 2개씩 저장... 그냥 다 저장하고 2곡씩 뽑아써야...
    
    for i in range(len(genres)):
        if genres[i] in genres_total_play:
            genres_total_play[genres[i]] += plays[i]
            genres_song_count[genres[i]].append((i, plays[i]))
        else:
            genres_total_play[genres[i]] = plays[i]
            genres_song_count[genres[i]] = [(i, plays[i])]
    
    popular_genre_list = sorted(genres_total_play.items(), key=lambda x:x[1], reverse=True)
    
    for i in popular_genre_list: # 이 순서대로 2곡씩
        genre_song_list = genres_song_count[i[0]]
        best_song_list = sorted(genre_song_list, key = lambda x:-x[1] )
        for ii in range(len(best_song_list)):
            #print(i[0], best_song_list[ii])
            answer.append(best_song_list[ii][0])
            if ii == 1:
                break

    return answer

고수의 파이썬 풀이

def solution(genres, plays):
    answer = []
    d = {e:[] for e in set(genres)}
    for e in zip(genres, plays, range(len(plays))):
        d[e[0]].append([e[1] , e[2]])
    genreSort =sorted(list(d.keys()), key= lambda x: sum( map(lambda y: y[0],d[x])), reverse = True)
    for g in genreSort:
        temp = [e[1] for e in sorted(d[g],key= lambda x: (x[0], -x[1]), reverse = True)]
        answer += temp[:min(len(temp),2)]
    return answer

현타 오지게 온다. ㅋㅋㅋ

파이썬 자료 구조를 더 자유롭게 쓸 수 있어야 알고리즘 문제를 더 쉽게 풀 수 있을 것 같은데...