컨트롤 제트 프로그래머스 코딩테스트 JAVA

컨트롤 제트 프로그래머스 코딩테스트 JAVA

 

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

프로그래머스

 

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

 

class Solution {
    public int solution(String s) {
        int answer = 0;
        String[] strarr = s.split(" ");
        int currentNum = 0;
        boolean subtractNext = false;
        
        
        for( String token : strarr ){
            if(token.equals("Z")) {
                answer -= currentNum;
            } else {
                int num = Integer.parseInt(token);
                answer +=  num;
                currentNum =  num;
                subtractNext = true;
            }
        }
        return answer;
    }
}

 

---

다른사람 풀이 : 스택 자료구조를 이용한 멋진 풀이라서 가져와봄! 배우자!

import java.util.*;

class Solution {
    public int solution(String s) {
        int answer = 0;
        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;
    }
}