728x90
반응형

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42579

 

코딩테스트 연습 - 베스트앨범

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

programmers.co.kr

 

문제 설명

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

속한 노래가 많이 재생된 장르를 먼저 수록합니다.

장르 내에서 많이 재생된 노래를 먼저 수록합니다.

장르 내에서 재생 횟수가 같은 노래 중에서는 고유 번호가 낮은 노래를 먼저 수록합니다.

노래의 장르를 나타내는 문자열 배열 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]번 노래를 그다음에 수록합니다.

약 45분만에 풀었다.

레벨 3을 처음 풀어봤는데 확실히 레벨2보다는 테크니컬적인 부분도 많이 요구하는 것 같다.

어떻게 풀어야할지 감조차 안잡히는 문제라면 참 머리아팠을 것 같은데 실마리가 보여서 차근차근 쫓아가며 풀 수 있었기 때문에 재미도 조금 있었던 것 같다.

 

from collections import defaultdict
def solution(genres, plays):
    answer = []
    genres_rank = []
    info = defaultdict(list)
    d = defaultdict(int)
    for i, j in enumerate(genres):
        d[j] += plays[i]
        info[j] += [(i, plays[i])]
    for key, value in sorted(d.items(), reverse=True, key=lambda item:item[1]):
        genres_rank.append(key)
    for genre in genres_rank:
        info[genre].sort(reverse=True, key=lambda x: x[1])
    for genre in genres_rank:
        for i in range(min(len(info[genre]) , 2)):
            answer.append(info[genre][i][0])
        
    return answer
​

// Test Case
genres = ["classic", "pop","classic", "classic", "pop","kpop","rock","rock"]
plays = [500, 600, 150, 800, 2500,1700,500,500]

genres_rank 리스트는 인기 많은 장르 순으로 장르의 이름만 따놓는 리스트이다. ['pop', 'kpop', 'classic', 'rock']

info 는 기본적인 모든 정보를 저장하는 리스트이다. 장르별로 분류하고 재생횟수와 고유번호까지 저장해놓는다.

아래는 info의 정렬전후 출력 결과이다.

 

defaultdict(<class 'list'>, {'classic': [(0, 500), (2, 150), (3, 800)], 'pop': [(1, 600), (4, 2500)], 'kpop': [(5, 1700)], 'rock': [(6, 500), (7, 500)]})
defaultdict(<class 'list'>, {'classic': [(3, 800), (0, 500), (2, 150)], 'pop': [(4, 2500), (1, 600)], 'kpop': [(5, 1700)], 'rock': [(6, 500), (7, 500)]})

 

인기가 많은 장르더라도 장르별 TOP2까지만 보여주는 게 목적이기 때문에

2와 장르의 곡수 중 작은 값까지만 반복해서 결과 리스트에 추가해준다.

728x90
반응형
728x90
반응형

https://programmers.co.kr/learn/courses/30/lessons/42885/

 

코딩테스트 연습 - 구명보트

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다. 예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 5

programmers.co.kr

무인도에 갇힌 사람들을 구명보트를 이용하여 구출하려고 합니다. 구명보트는 작아서 한 번에 최대 2명씩 밖에 탈 수 없고, 무게 제한도 있습니다.

예를 들어, 사람들의 몸무게가 [70kg, 50kg, 80kg, 50kg]이고 구명보트의 무게 제한이 100kg이라면 2번째 사람과 4번째 사람은 같이 탈 수 있지만 1번째 사람과 3번째 사람의 무게의 합은 150kg이므로 구명보트의 무게 제한을 초과하여 같이 탈 수 없습니다.

구명보트를 최대한 적게 사용하여 모든 사람을 구출하려고 합니다.

사람들의 몸무게를 담은 배열 people과 구명보트의 무게 제한 limit가 매개변수로 주어질 때, 모든 사람을 구출하기 위해 필요한 구명보트 개수의 최솟값을 return 하도록 solution 함수를 작성해주세요.

제한사항

무인도에 갇힌 사람은 1명 이상 50,000명 이하입니다.

각 사람의 몸무게는 40kg 이상 240kg 이하입니다.

구명보트의 무게 제한은 40kg 이상 240kg 이하입니다.

구명보트의 무게 제한은 항상 사람들의 몸무게 중 최댓값보다 크게 주어지므로 사람들을 구출할 수 없는 경우는 없습니다.

입출력 예

people limit return
[70, 50, 80, 50] 100 3
[70, 80, 50] 100 3

내 풀이.

태우고 나가야하니까 태우는 순간 계속 빠진다는 점에서 pop과popleft를 쓰고 싶어 우선순위 큐인 데크로 문제를 풀었다.

 

 

from collections import deque
def solution(people, limit):
    answer = 0
    people.sort(reverse=True)
    q = deque(people)
    while q:
        temp = q.popleft()
        if q and q[-1] <= limit-temp:
            q.pop()
        answer += 1
    return answer

 

런너기법처럼

리스트의 머리부터 뛰면서 체크하는a 와 꼬리부터 뛰면서 체크하는b의 합을 가지고 체크하는 방식의 풀이가 있어 가져왔다.

 

def solution(people, limit) :
    answer = 0
    people.sort()

    a = 0
    b = len(people) - 1
    while a < b :
        if people[b] + people[a] <= limit :
            a += 1
            answer += 1
        b -= 1
    return len(people) - answer
728x90
반응형
728x90
반응형

문제 출처 : programmers.co.kr/learn/courses/30/lessons/42626

 

코딩테스트 연습 - 더 맵게

매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같

programmers.co.kr

문제 설명

매운 것을 좋아하는 Leo는 모든 음식의 스코빌 지수를 K 이상으로 만들고 싶습니다. 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 Leo는 스코빌 지수가 가장 낮은 두 개의 음식을 아래와 같이 특별한 방법으로 섞어 새로운 음식을 만듭니다.

섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)

Leo는 모든 음식의 스코빌 지수가 K 이상이 될 때까지 반복하여 섞습니다.
Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville과 원하는 스코빌 지수 K가 주어질 때, 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수를 return 하도록 solution 함수를 작성해주세요.

제한 사항

  • scoville의 길이는 2 이상 1,000,000 이하입니다.
  • K는 0 이상 1,000,000,000 이하입니다.
  • scoville의 원소는 각각 0 이상 1,000,000 이하입니다.
  • 모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return 합니다.

입출력 예

scovilleKreturn

[1, 2, 3, 9, 10, 12] 7 2

입출력 예 설명

  1. 스코빌 지수가 1인 음식과 2인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
    새로운 음식의 스코빌 지수 = 1 + (2 * 2) = 5
    가진 음식의 스코빌 지수 = [5, 3, 9, 10, 12]

  2. 스코빌 지수가 3인 음식과 5인 음식을 섞으면 음식의 스코빌 지수가 아래와 같이 됩니다.
    새로운 음식의 스코빌 지수 = 3 + (5 * 2) = 13
    가진 음식의 스코빌 지수 = [13, 9, 10, 12]

모든 음식의 스코빌 지수가 7 이상이 되었고 이때 섞은 횟수는 2회입니다

import heapq
def solution(scoville, K):
    answer = 0
    heapq.heapify(scoville)
    if scoville[0] >= K:
        return 0
    while len(scoville)>=2:
        if scoville[0] >= K:
            break
        else:
            new = heapq.heappop(scoville) + 2* heapq.heappop(scoville)
            heapq.heappush(scoville, new)
            answer += 1
    if scoville[0] < K:
        return -1
    else:return answer
728x90
반응형
728x90
반응형

문제 설명

트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 트럭은 1초에 1만큼 움직이며, 다리 길이는 bridge_length이고 다리는 무게 weight까지 견딥니다.
※ 트럭이 다리에 완전히 오르지 않은 경우, 이 트럭의 무게는 고려하지 않습니다.

예를 들어, 길이가 2이고 10kg 무게를 견디는 다리가 있습니다. 무게가 [7, 4, 5, 6]kg인 트럭이 순서대로 최단 시간 안에 다리를 건너려면 다음과 같이 건너야 합니다.

경과 시간다리를 지난 트럭다리를 건너는 트럭대기 트럭

0 [] [] [7,4,5,6]
1~2 [] [7] [4,5,6]
3 [7] [4] [5,6]
4 [7] [4,5] [6]
5 [7,4] [5] [6]
6~7 [7,4,5] [6] []
8 [7,4,5,6] [] []

따라서, 모든 트럭이 다리를 지나려면 최소 8초가 걸립니다.

solution 함수의 매개변수로 다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight, 트럭별 무게 truck_weights가 주어집니다. 이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return 하도록 solution 함수를 완성하세요.

제한 조건

  • bridge_length는 1 이상 10,000 이하입니다.
  • weight는 1 이상 10,000 이하입니다.
  • truck_weights의 길이는 1 이상 10,000 이하입니다.
  • 모든 트럭의 무게는 1 이상 weight 이하입니다.

입출력 예

bridge_lengthweighttruck_weightsreturn

 

2 10 [7,4,5,6] 8
100 100 [10] 101
100 100 [10,10,10,10,10,10,10,10,10,10] 110

 

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42583

 

코딩테스트 연습 - 다리를 지나는 트럭

트럭 여러 대가 강을 가로지르는 일 차선 다리를 정해진 순으로 건너려 합니다. 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 알아내야 합니다. 트럭은 1초에 1만큼 움직이며, 다리 길이

programmers.co.kr

from collections import deque
def solution(bridge_length, weight, truck_weights):
    answer = 0
    q = deque(truck_weights)
    on = deque([0] * bridge_length)
    on_weight = 0
    while q :
        answer += 1
        down = on.popleft()
        on_weight -= down
        
        if q[0] + on_weight > weight:
            on.append(0)
        else:
            x = q.popleft()
            on.append(x)
            on_weight += x
    answer += bridge_length
    return answer
728x90
반응형
728x90
반응형

문제 설명

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 100%일 때 서비스에 반영할 수 있습니다.

또, 각 기능의 개발속도는 모두 다르기 때문에 뒤에 있는 기능이 앞에 있는 기능보다 먼저 개발될 수 있고, 이때 뒤에 있는 기능은 앞에 있는 기능이 배포될 때 함께 배포됩니다.

먼저 배포되어야 하는 순서대로 작업의 진도가 적힌 정수 배열 progresses와 각 작업의 개발 속도가 적힌 정수 배열 speeds가 주어질 때 각 배포마다 몇 개의 기능이 배포되는지를 return 하도록 solution 함수를 완성하세요.

제한 사항

작업의 개수(progresses, speeds배열의 길이)는 100개 이하입니다.

작업 진도는 100 미만의 자연수입니다.

작업 속도는 100 이하의 자연수입니다.

배포는 하루에 한 번만 할 수 있으며, 하루의 끝에 이루어진다고 가정합니다. 예를 들어 진도율이 95%인 작업의 개발 속도가 하루에 4%라면 배포는 2일 뒤에 이루어집니다.

 

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42586

 

코딩테스트 연습 - 기능개발

프로그래머스 팀에서는 기능 개선 작업을 수행 중입니다. 각 기능은 진도가 100%일 때 서비스에 반영할 수 있습니다. 또, 각 기능의 개발속도는 모두 다르기 때문에 뒤에 있는 기능이 앞에 있는

programmers.co.kr

풀이 1: 덱으로 풀기

from collections import deque
import math
def solution(progresses, speeds):
    answer = []
    q = []
    for i in range(len(progresses)):
        q.append(math.ceil((100-progresses[i]) / speeds[i]))
    q = deque(q)
    while q:
        cnt = 1
        x = q.popleft()
        temp = 0
        for i in q:
            if i<=x:
                temp += 1
            else: break
        cnt += temp
        for _ in range(temp):
            q.popleft()
        answer.append(cnt)
    return answer

 

풀이2: 덱 없이 리스트에서 수를 전처리 시켜두고 풀기

import math
def solution(progresses, speeds):
    answer = []
    cnt = 1
    q = []
    for i in range(len(progresses)):
        q.append(math.ceil((100-progresses[i]) / speeds[i]))

    for i in range(1, len(q)):
        if q[i-1] >= q[i]:
            q[i] = q[i-1]
    
    for i in range(1, len(q)):        
        if q[i-1]>=q[i]:
            cnt += 1
        else:
            answer.append(cnt)
            cnt = 1
    answer.append(cnt)
    return answer
728x90
반응형
728x90
반응형

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/42584

코딩테스트 연습 - 주식가격

solution.c 1 #include <stdio.h> 2 #include <stdbool.h> 3 #include <stdlib.h> 4 ​ 5 // prices_len은 배열 prices의 길이입니다. 6 int* solution ( int prices [], size_t prices_len ) { 7 // return 값은 malloc 등 동적 할당을 사용해주세요. 할당 길이는 상황에 맞게 변경해주세요. 8 int* answer = ( int* ) malloc ( 1 ); 9 return answer ; 10 } 실행 결과 ...

programmers.co.kr

문제 설명

초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때, 가격이 떨어지지 않은 기간은 몇 초인지를 return 하도록 solution 함수를 완성하세요.

제한사항

prices의 각 가격은 1 이상 10,000 이하인 자연수입니다.

prices의 길이는 2 이상 100,000 이하입니다.

입출력 예

prices

return

[1, 2, 3, 2, 3]

[4, 3, 1, 1, 0]

비교적 빠른 시간 안에 아이디어를 잡고 구현까지 성공했다.

큐/스택 카테고리에 들어가있던 문제이지만 큐, 스택과는 관계없이 대소관계를 비교하며

이중 반복문으로 풀 수 있었다.

def solution(prices):
    answer = []
    for i in range(len(prices)):
        cnt = 0
        for j in range(i+1, len(prices)):
            cnt +=1
            if prices[i] > prices[j] :
                break
        answer.append(cnt)
    return answer

큐와 스택을 이용해서 푸는 풀이는 없나하고 다른 사람의 문제 풀이를 참조해봤지만

돌아가는 구조 자체는 위와 크게 다르지 않았다.

prices를 deque() 를 이용해서 덱으로 만든 다음 popleft 를 이용해 맨 왼쪽 값을 빼내는 것 정도.

큐/스택은 문제풀이에서 능수능란하게 다루기 어려운 것 같다.

728x90
반응형
728x90
반응형

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/49993

 

코딩테스트 연습 - 스킬트리

 

programmers.co.kr

문제 설명

선행 스킬이란 어떤 스킬을 배우기 전에 먼저 배워야 하는 스킬을 뜻합니다.

예를 들어 선행 스킬 순서가 스파크 → 라이트닝 볼트 → 썬더일때, 썬더를 배우려면 먼저 라이트닝 볼트를 배워야 하고, 라이트닝 볼트를 배우려면 먼저 스파크를 배워야 합니다.

위 순서에 없는 다른 스킬(힐링 등)은 순서에 상관없이 배울 수 있습니다. 따라서 스파크 → 힐링 → 라이트닝 볼트 → 썬더와 같은 스킬트리는 가능하지만, 썬더 → 스파크라이트닝 볼트 → 스파크 → 힐링 → 썬더와 같은 스킬트리는 불가능합니다.

선행 스킬 순서 skill과 유저들이 만든 스킬트리1를 담은 배열 skill_trees가 매개변수로 주어질 때, 가능한 스킬트리 개수를 return 하는 solution 함수를 작성해주세요.

제한 조건

스킬은 알파벳 대문자로 표기하며, 모든 문자열은 알파벳 대문자로만 이루어져 있습니다.

스킬 순서와 스킬트리는 문자열로 표기합니다.

예를 들어, C → B → D 라면 CBD로 표기합니다

선행 스킬 순서 skill의 길이는 1 이상 26 이하이며, 스킬은 중복해 주어지지 않습니다.

skill_trees는 길이 1 이상 20 이하인 배열입니다.

skill_trees의 원소는 스킬을 나타내는 문자열입니다.

skill_trees의 원소는 길이가 2 이상 26 이하인 문자열이며, 스킬이 중복해 주어지지 않습니다.

입출력 예

skill

skill_trees

return

"CBD"

["BACDE", "CBADF", "AECB", "BDA"]

2

정해진 스킬트리의 순서를 인식하고, 그 스킬트리의 순서에 맞게 스킬을 찍고 있는지를 확인하기 위해

skill문자열을 덱으로 바꿔서 popleft() 를 쓸 수 있게 했다.

저렇게 풀다가 막힌 부분이 CBD의 스킬트리에서 D까지 스킬을 다 찍었을 때 q에 더 이상 요소가 남아있지 않아 popleft()실행 시 IndexError가 발생한다는 점이었는데,

try except문을 사용함과 동시에 스킬트리 순서를 지킨 개수를 answer에 하나씩 더하는 것보다

시작과 동시에 주어진 skill_trees의 개수를 answer에 넣어두고 스킬트리 순서를 지키지 않은 개수를 answer에서 하나씩 빼는 방법으로 접근했다. 코드모양이 조금 길어진 것 같지만 다른 접근방법으로 해결할 수 있어서 기뻤다.

 

from collections import deque
def solution(skill, skill_trees):
    answer = len(skill_trees)
    for tree in skill_trees:        
        q = deque(skill)
        s = q.popleft()
        for v in tree:
            if v in q:
                answer -= 1
                break
            elif s == v:
                try:
                    s = q.popleft()
                except IndexError:
                    break
    return answer
728x90
반응형
728x90
반응형

문제출처 : https://programmers.co.kr/learn/courses/30/lessons/42587

 

코딩테스트 연습 - 프린터

일반적인 프린터는 인쇄 요청이 들어온 순서대로 인쇄합니다. 그렇기 때문에 중요한 문서가 나중에 인쇄될 수 있습니다. 이런 문제를 보완하기 위해 중요도가 높은 문서를 먼저 인쇄하는 프린

programmers.co.kr

똑같은 백준 문제 : https://www.acmicpc.net/problem/1966

 

1966번: 프린터 큐

첫 줄에 test case의 수가 주어진다. 각 test case에 대해서 문서의 수 N(100이하)와 몇 번째로 인쇄되었는지 궁금한 문서가 현재 Queue의 어떤 위치에 있는지를 알려주는 M(0이상 N미만)이 주어진다. 다음

www.acmicpc.net

def solution(priorities, location):
    answer = 0
    priorities_copy = [0 for _ in range(len(priorities))]
    priorities_copy[location] = 1 
    while True:
        if priorities[0] == max(priorities):
            answer += 1
            if not priorities_copy[0]:
                del priorities[0] 
                del priorities_copy[0]
            else:break     
        else:
            priorities.append(priorities[0])
            priorities_copy.append(priorities_copy[0])
            del priorities_copy[0]
            del priorities[0]
    return answer

 

다른 사람의 풀이 보기를 해봤더니

python 내장함수인 any함수를 이용해서 푼 신기한 풀이가 있길래 공부가 되었다.

 

any함수를 이용한 풀이코드

def solution(priorities, location):
    queue =  [(i,p) for i,p in enumerate(priorities)]
    answer = 0
    while True:
        cur = queue.pop(0)
        if any(cur[1] < q[1] for q in queue):
            queue.append(cur)
        else:
            answer += 1
            if cur[0] == location:
                return answer

 

python any 함수는 반복 가능한(iterable) 자료형 x를 입력 인수로 받으며 이 x의 요소 중 하나라도 참이 있으면 True를 돌려주고, x가 모두 거짓일 때에만 False를 돌려주는 함수이다. 파이썬의 내장 함수이다.

 

>>> any([0, ""])
False
>>> any([])
False
>>> any([1,2,3,0])
True
728x90
반응형
728x90
반응형

문제 출처 : https://programmers.co.kr/learn/courses/30/lessons/12899

 

코딩테스트 연습 - 124 나라의 숫자

 

programmers.co.kr

문제 설명

124 나라가 있습니다. 124 나라에서는 10진법이 아닌 다음과 같은 자신들만의 규칙으로 수를 표현합니다.

124 나라에는 자연수만 존재합니다.

124 나라에는 모든 수를 표현할 때 1, 2, 4만 사용합니다.

예를 들어서 124 나라에서 사용하는 숫자는 다음과 같이 변환됩니다.

10진법

124 나라

10진법

124 나라

1

1

6

14

2

2

7

21

3

4

8

22

4

11

9

24

5

12

10

41

자연수 n이 매개변수로 주어질 때, n을 124 나라에서 사용하는 숫자로 바꾼 값을 return 하도록 solution 함수를 완성해 주세요.

제한사항

n은 500,000,000이하의 자연수 입니다.

 

 

def solution(n):
    answer = ''
    answer_list = ["4", "1", "2"]
    while n>0:
        answer = answer_list[n%3] + answer
        if not n%3:
            n = n//3 - 1
        else:
            n //= 3
    
    return answer
728x90
반응형
728x90
반응형

문제 출처 https://programmers.co.kr/learn/courses/30/lessons/42578

 

문제 설명

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

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

제한사항

clothes의 각 행은 [의상의 이름, 의상의 종류]로 이루어져 있습니다.

스파이가 가진 의상의 수는 1개 이상 30개 이하입니다.

같은 이름을 가진 의상은 존재하지 않습니다.

clothes의 모든 원소는 문자열로 이루어져 있습니다.

모든 문자열의 길이는 1 이상 20 이하인 자연수이고 알파벳 소문자 또는 '_' 로만 이루어져 있습니다.

스파이는 하루에 최소 한 개의 의상은 입습니다.

 

휴 40분 걸렸다

수학문제에 더 가까운 것 같은 문제.

의상카테고리가 하나일 때는 그 카테고리에 있는 개수를 출력하기만 하면 되지만

카테고리의 개수가 3개 이상으로 점점 늘어날수록 한 카테고리의 의상은 아예 선택하지 않는 경우까지 고려해야하다보니 구현을 어떻게 해야할지 통 감이 안와서 시간이 많이 걸렸다.

딕셔너리랑 각 카테고리별 요소 개수를 저장하는 리스트구현까지는 성공했는데

그 이후에 답을 얻어내는 과정을 모르겠어서 결국 답을 치팅했다.

(요소개수 + 1) 의 모든 곱이라니..

치팅하고 나서 공식으로 풀어보니까 위와 같이 기가막히게 똑같이 나왔다.

그래도 딕셔너리 밸류에 리스트를 넣는 등 테크니컬적인 부분에서는 공부가 됐다.

 

def solution(clothes):
    from collections import defaultdict
    d = defaultdict(list)
    def multi(arr):
        ans = 1
        for i in arr:
            ans *= (i+1)
        return ans
    for i in range(len(clothes)):
        d[clothes[i][1]] += [clothes[i][0]]
    lengths = [len(v) for v in d.values()]
    if len(lengths)==1:
        return lengths[0]
    else: return (multi(lengths) -1)
728x90
반응형

+ Recent posts