숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s
가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.
s
의 길이 ≤ 200s
의 원소 중 숫자 < 1,000s
는 숫자, "Z", 공백으로 이루어져 있습니다.s
에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.s
는 "Z"로 시작하지 않습니다.s
의 시작과 끝에는 공백이 없습니다.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 |
한 쪽 끝에서만 자료를 넣고 뺄 수 있는 LIFO(Last In First Out) 형식의 자료 구조
Stack의 연산
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;
}
}