Programmers_JAVA_컨트롤 제트

문제 설명

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

---

제한사항

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

---

ArrayList에 더한 값들을 스택처럼 관리하였다.  Stack 컬렉션을 사용하면 더욱 편리하고 메모리 상 이득을 많이 보긴 한다.

import java.util.ArrayList;
class Solution {
    public int solution(String s) {
        int answer = 0;
        ArrayList<Integer> Z = new ArrayList<>();
        int Z_count = 0;

        //String을 String 배열로 변환
        String[] s_arr = s.split(" ");

        for(int i = 0; i<s_arr.length;i++){
            //요소 중 숫자라면
            if(!s_arr[i].equals("Z")){
		//숫자 더함
                answer+=Integer.parseInt(s_arr[i]);
                //더한 숫자 Z배열에 넣음
                Z.add(Integer.parseInt(s_arr[i]));
            }
            else{
                if(i!=0){
                	//Z배열의 (크기-1)의 인덱스를 빼주었다 -> 스택과 같은 기능
                    answer-=Integer.parseInt(Z.get(Z.size()-1).toString());
                    //빼준 인덱스 값 제거
                    Z.remove(Z.size()-1);
                }
            }
        }
        return answer;
    }
}