1. 더 맵게 [heap]

이렇게 하면 test case는 통과하지만, 효율성 테스트에서는 시간 초과,,, 결과는 대부분 런타임 에러,,,

힙을 안쓴 코드여서 그렇지 뭐

def solution(scoville, K):
    '''
    input: Leo가 가진 음식의 스코빌 지수를 담은 배열 scoville, 원하는 스코빌 지수 K
    task: 섞은 음식의 스코빌 지수 = 가장 맵지 않은 음식의 스코빌 지수 + (두 번째로 맵지 않은 음식의 스코빌 지수 * 2)
    output: 모든 음식의 스코빌 지수를 K 이상으로 만들기 위해 섞어야 하는 최소 횟수
    모든 음식의 스코빌 지수를 K 이상으로 만들 수 없는 경우에는 -1을 return

    '''
    answer = 0
    for i in range(10**10):
        scoville.sort()
        answer +=1
        scoville.append(scoville[0]+scoville[1])
        del scoville[0]
        del scoville[0]

        if scoville[0] >=K: return answer
    return -1

사실 min이 맨 위에 있는 minheap 구조를 만들면 좋겠지만, 계속 정렬하는 과정에서 오래걸릴 것 같당..

but, python에는 heapq 라는 라이브러리가 있었고..! 이를 이용해보자

import heapq
def solution(scoville, k):
    heap = []
    for num in scoville:
        heapq.heappush(heap, num)
        cnt = 0
        print(heap)
    while heap[0] < k:
        print(heap)
        try:
             heapq.heappush(heap, heapq.heappop(heap) + (heapq.heappop(heap) * 2))
        except IndexError:
            return -1
        cnt += 1

    return cnt
  1. 주식 가격 [queue]

문제 이해하는데 쫌 걸렸다,,

from collections import deque
def solution(prices):
    '''
    초 단위로 기록된 주식가격이 담긴 배열 prices가 매개변수로 주어질 때,
    가격이 떨어지지 않은 기간은 몇 초인지를 return
    '''
    answer = []
    prices = deque(prices)

    while prices:
        temp = prices.popleft() # 기준
        count = 0
        for price in prices:  # 나머지 숫자 하나씩 다 살펴보기
            count += 1
            if temp > price: break
        answer.append(count)
    return answer
  1. 다리를 지나는 트럭 [stack]

처음에는 이렇게 했는데, 이러면 한번에 2대 이상의 트럭이 지나가는 경우를 따지기가 어려웠당

def solution(bridge_length, weight, truck_weights):
    '''
    트럭은 1초에 1만큼 움직임!
    다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight,
    트럭별 무게 truck_weights가 주어집니다.
    이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return
    '''
    time = -1
    truck_weights.reverse()
    counts = len(truck_weights)
    #print(truck_weights)
    #print(time)
    while truck_weights:
        time += bridge_length
        truck_weights.pop()
        time +=1
        #print(truck_weights)
        #print(time)
        if len(truck_weights)>1 and truck_weights[-1]+truck_weights[-2]<=weight:
            truck_weights.pop()
        if len(truck_weights)==0: break
    return time

다른 분의 코드를 참고하여, on_bridge와 on_time을 추가한 코드를 만드렀당

def solution(bridge_length, weight, truck_weights):
    '''
    트럭은 1초에 1만큼 움직임!
    다리 길이 bridge_length, 다리가 견딜 수 있는 무게 weight,
    트럭별 무게 truck_weights가 주어집니다.
    이때 모든 트럭이 다리를 건너려면 최소 몇 초가 걸리는지 return
    '''
    time = 0
    on_bridge = []
    on_time = []

    while(truck_weights or on_bridge) :
        time += 1
        if(on_time) :
            if(on_time[0] + bridge_length == time) :
                on_bridge.pop(0)
                on_time.pop(0)
        if(truck_weights) :
            if(sum(on_bridge) + truck_weights[0] <= weight) :
                on_bridge.append(truck_weights.pop(0))
                on_time.append(time)

    return time
  1. 위장 [hash]
from itertools import product
def solution(clothes):
    '''
    스파이가 가진 의상들이 담긴 2차원 배열 clothes가 주어질 때
    서로 다른 옷의 조합의 수를 return
    '''
    clothes_dict = dict()
    for cloth in clothes:
        if cloth[1] in list(clothes_dict.keys()):
            temp = list(clothes_dict[cloth[1]])
            temp.append(cloth[0])
            clothes_dict[cloth[1]] = temp
        else:
            clothes_dict[cloth[1]] = [cloth[0]]
    answer = 1
    print(clothes_dict)
    for items in list(clothes_dict.keys()):
        answer*= (len(clothes_dict[items])+1)
    answer -= 1 # 하나도 안 입었을 때
    return answer