숫자와 "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. 입력된 값들을 구분하기 위해 StringTokenizer를 사용해야겠다
2. 조건문을 통해 nextToken()값들을 더할때 다음 값이 Z이면 현재의 token값을 빼준다.
import java.util.StringTokenizer;
class Solution {
public int solution(String s) {
int answer = 0;
int pre= 0;
StringTokenizer st =new StringTokenizer(s);
while(st.hasMoreTokens()){
String token = st.nextToken();
if(token.equals("Z")){
answer-=pre;
}else{
int now =Integer.parseInt(token);
answer+=now;
pre=now;
}
}
return answer;
}
}🐴 풀이
1. 이전 값을 담을 pre 값을 생성해준다
2. StringTokenizer로 공백 기준으로 token들을 구분하여 담을수 있게 만든다
3. st가 token()값들이 더 있다면 while문 내부로 들어가게 한다
4. token값이 "Z" 라면 이전값을 뺴주고 아니라면 현재의 token()값을 더해준다.
5. String 타입의 s를 Integer.parseInt()로 정수형으로 변형
스택을 이용한 풀이도 가능했다.
import java.util.StringTokenizer;
import java.util.Stack;
class Solution {
public int solution(String s) {
int answer = 0;
Stack<Integer> stack = new Stack<>();
StringTokenizer st =new StringTokenizer(s);
while(st.hasMoreTokens()){
String token = st.nextToken();
if(token.equals("Z")){
stack.pop();
}else{
int num =Integer.parseInt(token);
stack.push(num);
}
}
for(int num:stack){
answer+=num;
}
return answer;
}
}🐴 풀이
1. 해당 token() 값이 "Z"라면 stack에서 pop()을 한다
2. "Z"가 아니라면 해당 token() 값을 stack에 push()한다
3. 향상된 for문을 통해 해당 stack에 있는 값들을 모두 더해 answer로 출력한다.
