[알고리즘] 컨트롤 제트

문제 설명

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

제한 사항

1 ≤ s의 길이 ≤ 200

-1,000 < s의 원소 중 숫자 < 1,000

s는 숫자, "Z", 공백으로 이루어져 있습니다.

s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.

연속된 공백은 주어지지 않습니다.

0을 제외하고는 0으로 시작하는 숫자는 없습니다.

s는 "Z"로 시작하지 않습니다.

s의 시작과 끝에는 공백이 없습니다.

"Z"가 연속해서 나오는 경우는 없습니다.

내가 풀이한 방법

class Solution {
    public int solution(String s) {
        int answer = 0;
        int prev = 0;
        String[] split = s.split(" ");

        for (String str : split) {
            if (str.equals("Z")) answer -= prev;
            else {
                int parseInt = Integer.parseInt(str);
                answer += parseInt;
                prev = parseInt;
            }
        }
        
        return answer;
    }
}

 

충격받은 풀이법

import java.util.*;

class Solution {
    public int solution(String s) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();

        for (String w : s.split(" ")) {
            if (w.equals("Z")) {
                stack.pop();
            } else {
                stack.push(Integer.parseInt(w));
            }
        }
        for (int i : stack) {
            answer += i;
        }
        return answer;
    }
}

 

스택을 이용한 풀이방법... 스택에 대해서 다시 한번 생각하게 되었다...