[프로그래머스] 컨트롤 제트

allnight5·2022년 11월 21일
0

프로그래머스 입문

목록 보기
34/53

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

def solution(s):
    answer = 0 
    stack =[]
    for i in s.split(" "):
        stack.append(i)
        if i == 'Z':
            stack.pop() 
            #리스트가 비어있지 않다면
            #리스트에서 내용을 꺼낸다
            if len(stack)>0:
                stack.pop() 
    for i in stack:
        answer += int(i)
    return answer
profile
공부기록하기

0개의 댓글