[프로그래머스][JAVA]Level 0 : 컨트롤 제트

문제 설명

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

 

제한사항

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

 

 

입출력 예sresult

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

 

 

 

---

 

문자열s의 처음과 끝은 공백이 아니니 split()을 이용하여 배열에 담는건 맞았는데,

단순하게 Z면 빼주고, 아니면 더하는 방식을 써버리니 Z가 연속일때 정상적으로 돌아가지가 않아 

테스트케이스 2번 7번만 오답이었다. ( 예외 케이스 : 1 2 3 Z Z / 결과 : 1 )

 

Z가 연속일때 어떻게 하지 하다가 그냥 Stack 이용해서 풀었더니 쉽게 풀렸다.

 

 

 

 

import java.util.Stack;
public class Solution {
    public int solution(String s) {
        int answer = 0;
        String[] arr=s.split(" "); // 공백을 기준으로 배열에 저장
        Stack<Integer> stack=new Stack<>();
        for(int i=0; i<arr.length; i++) {
            if (!arr[i].equals("Z")) { // "Z"가 아니면 
                stack.push(Integer.parseInt(arr[i]));
            } else { // "Z"이면서 
                if(!stack.isEmpty()) // stack이 빈공간이 아닐때 pop()
                    stack.pop();
            }
        }
        for(int i=0; i<stack.size(); i++){
            answer+=stack.get(i);
        }
        return answer;
    }