Programmers 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"가 연속해서 나오는 경우는 없습니다.

---

입출력 예sresult

s	result
"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 합니다.

입출력 예 #4, #5

설명 생략

 

 

※ 공지 - 2022년 11월 30일 제한사항 및 테스트 케이스가 수정되었습니다.

import java.util.Stack;
class Solution {
    public int solution(String s) {
        int answer = 0;
        Stack<Integer> stack = new Stack<>();
        for(String t : s.split(" ")) {
            if(t.equals("Z")) {
                if(!stack.isEmpty()) {
                    stack.pop();
                }
            }else {
                stack.push(Integer.parseInt(t));
            }
        }
        while(!stack.isEmpty()) answer += stack.pop();
        return answer;
    }
}

 

문자열을 공백으로 구분하여 배열로 만든다음 반복한다.   문자열이 "Z"가 아니면   Stack(LIFO : Last In First Out, 후입 선출구조) 에 int로 변경해 넣는다.   문자열이 "Z"이면 Stack의 맨위의 값을 꺼내 버린다.   Stack의 값을 모두 더해서 반환한다.

 

import java.util.LinkedList;
class Solution {
    public int solution(String s) {
        LinkedList<Integer> list = new LinkedList<>();
        for(String t : s.split(" ")) {
            if(t.equals("Z")) {
                list.removeLast();
            }else {
                list.add(Integer.parseInt(t));
            }
        }
        return list.stream().mapToInt(Integer::intValue).sum();
    }
}

 

위의 풀이와 같은데 LinkedList를 사용하였다.   "Z"이면 마지막 데이터를 지웠다.   합계를 Stream으로 구했다.   위에서 구한 stack의 합도 Stream의 sum()을 이용할 수 있다.   stack.stream().mapToInt(Integer::intValue).sum();