[연습문제]컨트롤 제트

더 많은 코드 (https://github.com/EunGyeongKim/TIL)

GitHub - EunGyeongKim/TIL: Today I Learne

문제 링크

프로그래머스

1. 문제 설명

숫자들이 공백으로 구분된 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 “Z”가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 “Z”로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.

---

2. 제한사항

• 0 < s의 길이 < 1,000
• 1,000 < s의 원소 중 숫자 < 1,000
• s는 숫자, "Z", 공백으로 이루어져 있습니다.
• s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
• 연속된 공백은 주어지지 않습니다.
• 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
• s의 시작과 끝에는 공백이 없습니다.
• 모든 숫자를 지우는 경우는 주어지지 않습니다.
• 지울 숫자가 없는 상태에서 "Z"는 무시합니다.

---

3. 입출력 예

s	result
"1 2 Z 3"	4
"10 20 30 40"	100
"10 Z 20 Z 1"	1

---

4. 코드

def solution(s):
    s = s.split(" ")
    stack_tmp = []

    for i in s:
        if str.isdigit(i):
            stack_tmp.append(int(i))
        elif i == 'Z' and len(stack_tmp) > 0:
            stack_tmp.pop(-1)
        # str.isdigit은 입력되는 수가 음수일 때 False로 판별하기 때문
        elif int(i) < 0 :
            stack_tmp.append(int(i))

    return sum(stack_tmp)