파이썬 알고리즘 문제 풀이

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

gogi masidda 2024. 1. 1. 16:23
  • 컨트롤 제트
문제 설명

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

def solution(s):
    answer = 0
    inputs = list(s.split())
    N = len(inputs)    
    for i in range(N):
        if inputs[i] == "Z":
            answer -= int(inputs[i-1])
        else:
            answer += int(inputs[i])
    return answer

처음에 isdigit()로 문자가 숫자인지 판별해서 하려고 했는데 isdigit는 음수가 나오면 false를 반환하는 문제가 있었다.

728x90