백준[node.js] > 1541번 잃어버린 괄호

문제

https://www.acmicpc.net/problem/1541

 

문제풀이

첫번째 for문에서 [ '+1' , '-2' , '-3' , '+2' , '+2' , '3' ] 와 같은 arr를 생성하고 

두번째 for문에서 answer을 계산한다

 

핵심은 음수 이후에 연속적으로 양수가 나올 때 해당 양수는 결국 음수로 계산해야 최솟값을 구할 수 있다는 점이다.

예시) 1-2+3+3 => 1-(2+3+3) 

 

이런식으로 음수이후로 연속적으로 나타나는 양수는 모두 음수처리 후 모든 수들을 합하면 최솟값을 구할 수 있다.

let fs = require("fs");
let input = fs.readFileSync("/dev/stdin").toString().trim();

let num = "";
let numArr = [];
for (let i = 0; i < input.length; ++i) {
    if (i === 0) {
        if (input[i] === "-") {
            num += input[i];
        } else {
            num += `+${input[i]}`;
        }
    } else {
        if (input[i] === "+" || input[i] === "-") {
            numArr.push(num);
            num = input[i];
        } else {
            num += input[i];
        }
    }

    if (i === input.length - 1) {
        numArr.push(num);
    }
}

let answer = 0;
for (let i = 0; i < numArr.length; ) {
    let n = numArr[i];
    if (n[0] === "-") {
        let temp = -1 * Number(n);
        let idx = i + 1;
        while (idx < numArr.length && numArr[idx].at(0) === "+") {
            temp += Number(numArr[idx]);
            idx = idx + 1;
        }

        answer -= temp;
        i = idx;
    } else {
        answer += Number(n);
        ++i;
    }
}
console.log(answer);