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

컨트롤 제트

숫자와 "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 2 Z 3"	4
"10 20 30 40"	100
"10 Z 20 Z 1"	1
"10 Z 20 Z"	0
"-1 -2 -3 Z"	-3

 

입출력 예 #1

• 본문과 동일합니다.

입출력 예 #2

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

입출력 예 #3

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

 

import java.util.Arrays;
import java.util.Stack;

public class CtrlZ {

	public static void main(String[] args) {
		System.out.println(solution("10 Z 20 Z 1"));
	}
	
	public static int solution(String s) {
        int answer = 0;
//        String[] str = s.split(" ");
//
//        for(int i=0; i<str.length; i++) {
//        	if(str[i].equals("Z")) {
//        		answer-=Integer.valueOf(str[i-1]);
//        		continue;
//        	}
//        	answer+=Integer.valueOf(str[i]);
//        }
//        
        Stack<Integer> stack = new Stack<>();
        
        for(String w : s.split(" ")) {
        	if(w.equals("Z")) 
        		stack.pop();
        	else {
        		stack.push(Integer.parseInt(w));
        	}
        }
        
        for(int i : stack) {
        	answer += i;
        }
        
        return answer;
    }
}

두 가지 방법으로 풀어보았다.

 

첫 번째 방법은 입력된 문자열 s를 split(" ")을 이용해 공백 기준으로 String배열에 담고

배열을 순회하면서 일단 숫자면 answer변수에 무조건 담고 if조건문으로 문자열 "Z"를 만났을 때 바로 전에 있던 인덱스의 숫자를 answer에서 빼주고 continue시켜주었다.

 

위의 방법보다 좀 더 좋은 풀이가 있을 것 같아서 다른 풀이를 보고 stack을 이용한 풀이도 참고해 보았다.

 

두 번째 방법은 stack의 특징을 이용한 것인데 "Z"를 만나지 않았으면 stack에 int형으로 push()해주고 "Z"를 만났으면

마지막으로 push했던 숫자를 pop()을 해서 stack의 특징인 Last In First Out을 통해 문자열 s를 stack에 저장했다.

 

향상 for문에서 배열 이외에 stack도 들어갈 수 있는 것과 두 번째 인자로 참조변수만 넣을 수 있는 줄 알았는데 s.split()처럼 바로 사용 가능한 줄 몰랐다.