문제: 코딩테스트 연습 - 컨트롤 제트 | 프로그래머스 스쿨 (programmers.co.kr)
문제설명
숫자와 "Z"가 공백으로 구분되어 담긴 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 "Z"가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 "Z"로 이루어진 문자열 s가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.
제한사항
- 1 ≤ s의 길이 ≤ 200
- -1,000 < s의 원소 중 숫자 < 1,000
- s는 숫자, "Z", 공백으로 이루어져 있습니다.
- s에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.
- 연속된 공백은 주어지지 않습니다.
- 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
- s는 "Z"로 시작하지 않습니다.
- s의 시작과 끝에는 공백이 없습니다.
- "Z"가 연속해서 나오는 경우는 없습니다.
입출력 예
s | result |
"1 2 Z 3" | 4 |
"10 20 30 40" | 100 |
"10 Z 20 Z 1" | 1 |
"10 Z 20 Z 1" | 0 |
"-1 -2 -3 Z" | -3 |
나의 문제 풀이
class Solution {
public int solution(String s) {
int answer = 0;
String[] arr = s.split(" ");
for (int i = 0; i < arr.length; i++) {
if(arr[i].equals("Z")){
answer -= Integer.parseInt(arr[i - 1]);
}else {
answer += Integer.parseInt(arr[i]);
}
}
return answer;
}
}
다른 사람의 문제풀이
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;
}
}
느낀점
Stack을 사용하기 좋은 문제라고 댓글에 적혀있었다. 나도 문제에 마다 무엇을 사용할지 바로바로 나올 수 있도록 연습해야겠다.
'코딩테스트 연습 > 프로그래머스 LV.0' 카테고리의 다른 글
[프로그래머스] Lv.0 숨어있는 숫자의 덧셈 (1) - 자바(Java) (0) | 2024.03.24 |
---|---|
[프로그래머스] Lv.0 소인수분해 - 자바(Java) (0) | 2024.03.23 |
[프로그래머스] Lv.0 최빈값 구하기 - 자바(Java) (0) | 2024.03.21 |
[프로그래머스] Lv.0 짝수는 싫어요 - 자바(Java) (0) | 2024.03.20 |
[프로그래머스] Lv.0 피자 나눠 먹기 (2) - 자바(Java) (0) | 2024.03.19 |