컨트롤 제트 - Python

def solution(s):
    command = list(s.split())
    answer = 0
    temp = 0
    for item in command:
        if item == 'Z':
            answer -= temp
        else:
            temp = int(item)
            answer += temp
    return answer


# Test Cases
print(solution("1 2 Z 3"))
print(solution("10 20 30 40"))
print(solution("10 Z 20 Z 1"))
print(solution("10 Z 20 Z"))
print(solution("-1 -2 -3 Z"))

  temp 변수를 선언하여 구현했습니다.

 

---

def solution(s):
    stack = []
    for a in s.split():
        if a != 'Z':
            stack.append(int(a))
        else:
            if stack:
                stack.pop()

    return sum(stack)

  그런데 다른 사람의 풀이를 보고 아차 싶네요. 참 잘 풀었다 싶습니다.

 

---

• 프로그래머스: https://school.programmers.co.kr/learn/courses/30/lessons/120853

프로그래머스