[프로그래머스_Java] Lv.0 컨트롤 제트

문제 설명

숫자와 "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[] arr = s.split(" ");
        
        for (int i = 0; i < arr.length; i++) {
            if (arr[i].equals("Z")) {
                answer -= prev;
            } else {
                answer += Integer.parseInt(arr[i]);
                prev = Integer.parseInt(arr[i]);
            }
        }
        return answer;
    }
}

• 문자열을 공백 기준으로 분리해 배열에 저장한다.
• 배열을 순회하며 현재 인덱스에 해당하는 값이 “Z”가 아닌 경우 현재 값을 저장하고, 더한다.
• 배열을 순회하며 현재 인덱스에 해당하는 값이 “Z”인 경우 이전 값(prev)을 빼준다.

다른 방법

import java.util.Stack;

class Solution {
    public int solution(String s) {
        int answer = 0;
        String[] arr = s.split(" ");
        Stack<Integer> stack = new Stack<>();

        for (String str : arr) {
            if (str.equals("Z")) {
                answer -= stack.pop();
            } else {
                int num = Integer.parseInt(str);
                answer += num;
                stack.push(num);
            }
        }
        return answer;
    }
}

• Stack을 활용할 수 있다.

 

 

프로그래머스