[백준] 슈퍼 마리오 (JS)

문제

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

슈퍼 마리오 앞에 10개의 버섯이 일렬로 놓여져 있다. 이 버섯을 먹으면 점수를 받는다.

슈퍼 마리오는 버섯을 처음부터 나온 순서대로 집으려고 한다. 하지만, 모든 버섯을 집을 필요는 없고 중간에 중단할 수 있다. 중간에 버섯을 먹는 것을 중단했다면, 그 이후에 나온 버섯은 모두 먹을 수 없다. 따라서 첫 버섯을 먹지 않았다면, 그 이후 버섯도 모두 먹을 수 없다.

마리오는 받은 점수의 합을 최대한 100에 가깝게 만들려고 한다.

버섯의 점수가 주어졌을 때, 마리오가 받는 점수를 출력하는 프로그램을 작성하시오.

입력

총 10개의 줄에 각각의 버섯의 점수가 주어진다. 이 값은 100보다 작거나 같은 양의 정수이다. 버섯이 나온 순서대로 점수가 주어진다.

출력

첫째 줄에 마리오가 받는 점수를 출력한다. 만약 100에 가까운 수가 2개라면 (예: 98, 102) 마리오는 큰 값을 선택한다.

풀이

1. 받은 점수의 합을 최대한 100에 가깝게 만들어야 한다. -> 누적합을 구한다.

2. 누적합이 100일 경우 그대로 100을 출력한다.

3. 누적합이 100을 초과했을 경우 differ1(이전 누적합, 100), differ2(현재 누적합, 100)의 크기를 비교하여 100에 더 가까운 누적합을 출력한다. (차이가 같을 경우 더 큰 숫자 출력)

4. 반복문이 끝난후에도 100을 넘기지 못했을 경우 현재 누적합이 100과 가장 가까운 숫자가 된다.

const fs = require("fs");
const input = fs
  .readFileSync("./dev/stdin")
  .toString()
  .trim()
  .split("\n")
  .map(Number);

Solution = (input) => {
    let prefixSum = 0;
    
    for (let i = 0; i < input.length; i++) {
        if (prefixSum === 100) break;
        else if (prefixSum + input[i] > 100) {
            const diffBefore = 100 - prefixSum;
            const diffAfter = (prefixSum + input[i]) - 100;
            if (diffBefore >= diffAfter) {
                prefixSum += input[i];
            }
            break;
        }
        prefixSum += input[i]; 
    }
    
    console.log(prefixSum);
}

Solution(input);

결론

누적합을 이용하여 구현하였다.

문제의 조건을 꼼꼼하게 읽으면 쉽게 해결할 수 있는 문제이다.