[프로그래머스(Java)] 컨트롤 제트 / Stack String.split()

 

Level. 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;
        String sum = "";
        for(int i=0; i<s.length(); i++){
            if(s.charAt(i) == ' '){
                if(s.charAt(i+1) == 'Z')
                    i+=2;
                else 
                    answer += Integer.parseInt(sum);
                sum = "";
            }
            else 
                sum += Character.toString(s.charAt(i));
        }
        
        if(sum.length() > 0) 
            answer += Integer.parseInt(sum);
        
        return answer;
    }
}

 

해결방법

1.  문자열을 반복문을 사용해 탐색한다.

2. 현재 문자가 숫자인 공백이 아닌 경우, 숫자이므로 Character.toString을 사용해 해당 문자를 sum 문자열에 저장한다.

3. 현재 문자가 공백인 경우, 공백 다음이 'Z'가 아니라면, answer에 sum을 더한다.

4. 공백 다음이  'Z'라면, answer에 sum을 더하는 작업은 생략하고, 'Z' 다음다음의 문자를 탐색한다. 

 

 

풀이2

import java.util.*;

class Solution {
    public int solution(String s) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();
        String[] str = s.split(" ");
        for(int i=0; i<str.length; i++){
            if(str[i].equals("Z")){
                stack.pop();
            }
            else {
                stack.push(Integer.parseInt(str[i]));
            }
        }
        
        for(int i: stack){
            answer += i;
        }
        
        return answer;
    }
}

 

사용한 자료구조

- Stack : 순서대로 탐색하여 이전 숫자를 더하거나, 더하지 않아야 하므로 후입선출인 스택이라는 자료구조를 사용하게 되었다. 

 

---

https://school.programmers.co.kr/learn/courses/30/lessons/120853?language=java 

프로그래머스