[프로그래머스] 고득점 kit - 스택 / 큐

<문제 링크>

https://school.programmers.co.kr/learn/courses/30/parts/12081

프로그래머스

<풀이 전략>

1. 같은 숫자는 싫어: 스택을 활용해서 풀었다.

2. 올바른 괄호: 마찬가지로 스택을 활용해서 풀었다

3. 기능 개발: 입력값을 큐로 변환해서 완료 될 때마다 하나씩 빼가면서 풀었다.

4. 프린터: enumerate랑 큐를 활용해서 요청 문서 번호가 나올 때까지 큐를 돌려서 풀었다.

5. 다리를 지나는 트럭: 다리를 주어진 길이에 맞는 큐로 구현해서 트럭의 이동을 구현해서 풀었다. 주의할 점으로는 현대 다리의 하중을 검사할 때 sum을 쓰면 효율성을 통과하지 못하므로 현재 하중을 따로 변수로 할당해서 계산해줘야 한다.

6. 주식 가격: 2중 for문으로 풀어도 통과가 된다. 주의할 점은 인덱스 등 범위 설정을 잘해줘야 된다.

<정답 코드>

# 1. 같은 숫자는 싫어

def solution(arr):
    answer = []
    for num in arr:
        if answer and answer[-1] == num:
            continue
        answer.append(num)

    return answer

 

# 2. 올바른 괄호

def solution(s):
    answer = True
    
    stack = []
    
    for char in s:
        
        if not stack: 
            stack.append(char)
            
        else:
            if stack[-1] == '(' and char == ')':
                stack.pop()
            else:
                stack.append(char)
    
    if stack:
        answer = False


    return answer

 

# 3. 기능개발

from collections import deque


def solution(progresses, speeds):

    progresses = deque(progresses)
    speeds = deque(speeds)

    answer = []
    
    while True:
        
        for i in range(len(progresses)):
            progresses[i] += speeds[i]

        if progresses[0] >= 100:

            done = 0

            while True:

                if not progresses:
                    break

                if progresses[0] < 100:
                    break

                progresses.popleft()
                speeds.popleft()

                done += 1

            answer.append(done)

        if not progresses:
            break

    return answer

 

# 4. 프린터

from collections import deque

def solution(priorities, location):
    printer = deque(enumerate(priorities))
    answer = 0


    while True:
        
        if printer[0][1] == max(printer, key=lambda x:x[1])[1]:

            answer += 1
            idx = printer.popleft()[0]

            if idx == location:
                break

        else:
            printer.append(printer.popleft())

            

    return answer

 

# 5. 다리를 지나는 트럭

from collections import deque


def solution(bridge_length, weight, truck_weights):
    answer = 1
    
    bridge = deque([0 for _ in range(bridge_length)])
    truck_weights = deque(truck_weights)
    truck_cnt = len(truck_weights)
    cnt = 0
    current_weight = 0

    while True:
        
        out = bridge.popleft()

        if out > 0:
            cnt += 1
            current_weight -= out
            if cnt == truck_cnt:
                break

        if truck_weights and weight >= current_weight + truck_weights[0]:
            in_truck = truck_weights.popleft()
            bridge.append(in_truck)
            current_weight += in_truck
        else:
            bridge.append(0)

        answer += 1    

    return answer

 

# 6. 주식가격

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