[프로그래머스] 컨트롤 제트

문제 설명

숫자와 "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"	0
"-1 -2 -3 Z"	-3

 

---

 

나의 코드

function solution(s) {
    let arr = s.split(" ");
    let stack = [];
    
    for (let i = 0; i < arr.length; i++) {
        if (arr[i] === "Z") {
            stack.pop();
        } else {
            stack.push(Number(arr[i]));
        }
    }
    
    return stack.reduce((acc, cur) => {
        return acc + cur;
    }, 0);
}

 

방식

1. 문자열을 공백 기준으로 나눈다.

2. 배열을 순회한다.

3. 스택을 이용해 값을 관리한다.

4. 반복문이 끝나면 스택에 남아 있는 숫자들만 합산한다.

---

 

다른 사람의 코드

function solution(s) {
    const stack = []

    s.split(' ').forEach((target) => {
        if(target === 'Z') stack.pop();
        else stack.push(+target)
    })

    return stack.length ? stack.reduce((pre, cur) => pre + cur) : 0;
}

 

나와의 차이점

나와 동일하게 스택을 활용한 풀이 방식이지만, for문 대신 forEach()를 사용해 배열 요소를 하나씩 처리하여 코드가 더 간결하고 깔끔함.