[백준 1744번] 수 묶기 - 자바스크립트(nodejs)

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

1744번: 수 묶기

const fs = require('fs');
const [n, ...input] = fs.readFileSync('../input.txt').toString().trim().split('\n').map(Number);

const negative = [];
const positive = [];
let zero = 0;
let answer = 0;

input.forEach((v) => {
  if (v < 0) {
    negative.push(v);
  } else if (v > 1) {
    positive.push(v);
  } else if (v === 0) {
    zero++;
  } else {
    answer++;
  }
});

negative.sort((a, b) => b - a);
positive.sort((a, b) => a - b);

while (positive.length) {
  if (positive.length === 1) {
    answer += positive.pop();
    break;
  }
  const value = positive.pop();
  const value2 = positive.pop();
  answer += value * value2;

  if (positive.length === 1) {
    answer += positive.pop();
    break;
  }
}

while (negative.length) {
  if (negative.length === 1) {
    if (zero !== 0) {
      zero--;
      break;
    } else {
      answer += negative.pop();
      break;
    }
  }
  if (negative.length % 2 === 1 && zero !== 0) {
    negative.shift();
    zero--;
  }
  const value = negative.pop();
  const value2 = negative.pop();
  answer += value * value2;

  if (negative.length === 1) {
    answer += negative.pop();
    break;
  }
}

console.log(answer);

1은 양수에 곱하면 더하는 값이 작아지므로 ex) 1, 11 => 곱하면 11 or 더하면 12 그냥 answer에 1을 더해준다.

음수 양수를 따로 배열에 저장해 sort로 정렬한 후 홀수, 짝수 개수를 생각하며 알고리즘을 구성한다.

 

만약 0이 있을 때는 음수가 홀수일 때 음수중에 가장 큰 값이랑 곱해줘야 합이 최대이다.

그리고, 음수, 양수가 1개씩만 있을 때를 생각해 알고리즘을 구상하면 쉽게 풀 수 있다.