문제 설명
숫자들이 공백으로 구분된 문자열이 주어집니다. 문자열에 있는 숫자를 차례대로 더하려고 합니다. 이 때 “Z”가 나오면 바로 전에 더했던 숫자를 뺀다는 뜻입니다. 숫자와 “Z”로 이루어진 문자열s
가 주어질 때, 머쓱이가 구한 값을 return 하도록 solution 함수를 완성해보세요.
제한사항
- 0 <
s
의 길이 < 1,000- -1,000 <
s
의 원소 중 숫자 < 1,000s
는 숫자, "Z", 공백으로 이루어져 있습니다.s
에 있는 숫자와 "Z"는 서로 공백으로 구분됩니다.- 연속된 공백은 주어지지 않습니다.
- 0을 제외하고는 0으로 시작하는 숫자는 없습니다.
s
의 시작과 끝에는 공백이 없습니다.- 모든 숫자를 지우는 경우는 주어지지 않습니다.
- 지울 숫자가 없는 상태에서 "Z"는 무시합니다.
입출력 예
s result "1 2 Z 3" 4 "10 20 30 40" 100 "10 Z 20 Z 1" 1
입출력 예 설명
입출력 예 #1
- 본문과 동일합니다.
입출력 예 #2- 10 + 20 + 30 + 40 = 100을 return 합니다.
입출력 예 #3- "10 Z 20 Z 1"에서 10 다음 Z, 20 다음 Z로 10, 20이 지워지고 1만 더하여 1을 return 합니다.
function solution(s) {
const sArr = s.split(' ')
for (let i = 0; i < sArr.length; i++ ) {
if (sArr[i] == "Z" && sArr[0] !== "Z") {
sArr.splice((i-1), 2)
i = i - 2;
}
if (sArr[i] == "Z" && sArr[0] === "Z") {
sArr.splice((i), 1)
i = i - 1;
}
}
return sArr.map(n => +n).reduce((a, c) => a + c, 0);
}
풀어내는 방법은 금방 생각했지만, 코드로 구현해내는 것에 많은 시간을 들였다.
for
문 안에 if
문을 두 가지로 나눈 이유는 다음과 같다.
일단 두 번째 if
문의 조건 중 sArr[0] === "Z"
이 있는 이유는 입력되는 s
가 "1 Z Z 2"와 같을 때를 고려했기 때문이다. 처음 나오는 Z는 1과 같이 지워지면서 문자열이 "Z 2"와 같게 바뀔 것이고, 이 때 Z 앞에 아무 것도 없기 때문에 두 개의 인덱스(Z와 바로 앞의 인덱스)를 지울 수 없게 되면서 오류가 생기게 된다. 이를 방지하기 위해 위와 같은 조건을 명시해두었다.
따라서 첫 번째 if
문의 조건 중 sArr[0] !== "Z"
이 있는 이유는 Z가 가장 앞에 존재하지 않을 때, Z와 그 앞의 인덱스를 지울 수 있게 하기 위해 명시해두었다. 그렇게 되면 배열을 순회하는 for
루프는 지워진 것들을 뛰어 넘어 제대로 배열을 순회할 수 없게 되기 때문에 i = i - 2
를 적어두었다.
문제를 풀어 나가면서 splice()
, slice()
, split()
의 차이점에 대해 정리해야겠다는 생각을 했다. 프로그래머스 Lv.0을 다 풀고 나서 한번 정리해서 올려야겠다.