Lv.0 컨트롤 제트

문제 설명

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

입출력

코드 풀이

def solution(s):
    arr = s.split(' ')
    result =[]
    for i in arr :
        if i=='Z':
            result.pop()
        else:
            result.append(int(i))
    return sum(result)

• 문자열 s를 공백을 기준으로 분리하고, 스택을 사용하여 "Z"가 나오면 바로 이전 숫자를 스택에서 제거하는 방식으로 동작

출처: 프로그래머스 코딩테스트 연습