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

문제 설명

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

제한사항

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

 

코드 1

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

            for(int i = 0;i<stt.length;i++){
                if(i!=0&&stt[i].equals("Z"))
                    answer-=temp;
                else if('9'>=stt[i].charAt(0) && '1'<=stt[i].charAt(0)){
                    answer+=Integer.valueOf(stt[i]);
                    temp = Integer.valueOf(stt[i]);
                }
                else{
                    stt[i] = stt[i].replace("-","");
                    answer-=Integer.valueOf(stt[i]);
                    temp = -Integer.valueOf(stt[i]);
                }
            }
            return answer;
        }
    }

 

 

코드 2

import java.util.Stack;

class Solution {
     public int solution(String s) {
        Stack<Integer> stack = new Stack<>();
        for (String str : s.split(" ")) {
            if(str.equals("Z"))
                stack.pop();
            else if(str.equals(" "))
                continue;
            else
                stack.push(Integer.valueOf(str));
        }
        return stack.stream().mapToInt(n -> n).sum();
    }
}

 

 

위 코드1은 스택 없이 구현한 코드이고, 코드2는 스택을 사용하여 구현한 코드이다.

하눈에 봐도 스택을 사용한 코드가 더 정결하고 간단하게 구현되었다.

 

 

 

 

 

출처: 프로그래머스 코딩 테스트 연습, https://school.programmers.co.kr/learn/challenges