프로그래머스 - 컨트롤 제트

문제

https://school.programmers.co.kr/learn/courses/30/lessons/120853

프로그래머스

 

 

 

풀이

스택을 이용해 문자열 s에 있는 숫자들을 관리한다.

문자열 s에 있는 숫자들을 차례대로 스택에 넣는다.

숫자가 아닌 Z 차례일 경우 스택에서 숫자를 pop 하면 바로 전에 더했던 숫자를 빼는 것과 같다.

마지막까지 스택에 남아있는 숫자들을 더하면 답이 된다.

 

 

 

코드

python

def solution(s):
    stack = []
    
    for n in s.split(" "):
        if "Z" == n:
            stack.pop()
        else:
            stack.append(int(n))
            
    answer = sum(stack)
    return answer