백준 1541 잃어버린 괄호 (그리디)

문제

세준이는 양수와 +, -, 그리고 괄호를 가지고 식을 만들었다. 그리고 나서 세준이는 괄호를 모두 지웠다.

그리고 나서 세준이는 괄호를 적절히 쳐서 이 식의 값을 최소로 만들려고 한다.

괄호를 적절히 쳐서 이 식의 값을 최소로 만드는 프로그램을 작성하시오.

제한 사항

첫째 줄에 식이 주어진다. 식은 ‘0’~‘9’, ‘+’, 그리고 ‘-’만으로 이루어져 있고, 가장 처음과 마지막 문자는 숫자이다. 그리고 연속해서 두 개 이상의 연산자가 나타나지 않고, 5자리보다 많이 연속되는 숫자는 없다. 수는 0으로 시작할 수 있다. 입력으로 주어지는 식의 길이는 50보다 작거나 같다.

입출력 예

풀이

const input = require('fs').readFileSync('/dev/stdin').toString().trim();
const sol = (input) => {
  const getSum = (arr) => {
    const answer = arr
      .split(/[\+-]/)
      .map(Number)
      .reduce((a, c) => a + c, 0);

    return answer;
  };
  const firstMinus = input.indexOf("-");

  if (firstMinus === -1) {
    return getSum(input);
  } else {
    const plus = getSum(input.slice(0, firstMinus));
    const minus = getSum(input.slice(firstMinus + 1));
    return plus - minus;
  }
};

// 마이너스 앞에 괄호를 치는것이 좋아보임.

console.log(sol(input));

특정 문자 기준 split을 하는 방법이 나와있다.

식의 값을 최소로 만들기 위해서는 뺄셈 연산자 뒤의 수를 크게 만들면 된다. -가 나오는 첫 인덱스를 찾고 그 앞과 뒤로 나눠서 합을 구하고 두개의 차를 return 한다.