[프로그래머스] 컨트롤 제트 자바(JAVA)

@notepad_jj2

츄르사려고 코딩하는 코집사입니다.

---

1. [프로그래머스] 컨트롤 제트 자바(JAVA)

1) 문제 제목 : 컨트롤 제트

 

2) 문제 출처

https://school.programmers.co.kr/learn/courses/30/lessons/120853

프로그래머스

 

2. 문제

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

 

3. 제약사항

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

 

4. 입출력

sresult

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

---

입출력 예 설명

입출력 예 #1

• 본문과 동일합니다.

입출력 예 #2

• 10 + 20 + 30 + 40 = 100을 return 합니다.

입출력 예 #3

• "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.

 

5. 풀이

- 처음에는 배열에서 작업을 하려고 했는데, 그렇게 하면 더 복잡해져서 스택을 사용했다.

- split으로 분리한 다음, 배열의 크기만큼 반복하여 Z가 아니면 stack에 push를 하고, Z면 stack에서 값을 꺼낸 다음 stack에 있는 값들을 다 더하면 된다.

 

6. 소스 코드

import java.util.Stack;

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

---