[Baekjoon]20186번 수 고르기 - Javascript

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

20186번: 수 고르기

문제

N개의 자연수가 좌우로 배열되어 있다. 여러분은 이 중 K개를 골라야 한다. 고를 때는 K개 모두를 한번에 골라야 한다.

여러분이 고른 수 각각에 대해서 그 수의 점수를 계산할 것이다. 점수는 자신의 수에서 자신의 왼쪽에 있는 수 중 선택된 수의 개수를 뺀 값이다. 이렇게 고른 수 각각에 그 점수를 계산한 후 전체점수는 계산된 점수들의 합이다. 여러분은 전체점수가 최대가 되도록 K개의 수를 골라야 한다.

예를 들어서, N = 5개의 자연수가 순서대로 2 3 1 2 1 로 주어지고, K = 3이라고 하자. 만약 첫 번째 2와 두 개의 1을 골랐다면, 각 수의 점수를 계산해서 나열하면 2 0 −1과 같다. 따라서 전체 점수는 1이다. 만약 첫 번째 2와 3, 그리고 두 번째 2를 고르고, 각 수의 점수를 계산해서 나열하면, 2 2 0과 같다. 따라서 전체점수는 4이다. 이 예에서 전체점수의 최댓값은 4이다.

N개의 자연수 배열과 양의 정수 K가 주어질 때, 전체점수의 최댓값을 출력하는 프로그램을 작성하시오.

입력

첫 번째 줄에 N과 K가 공백 하나를 사이로 두고 주어진다. 두 번째 줄에 N개의 자연수가 공백 하나를 사이에 두고 주어진다.

출력

첫 번째 줄에 주어진 N개의 수 중 K개의 수를 고를 때, 전체점수의 최댓값을 출력한다.

제한

• 1 ≤ N ≤ 5 000
• 1 ≤ K ≤ N
• 주어지는 자연수는 1 이상 100 000 이하

서브태스크

번호배점제한

1	4	N ≤ 3
2	25	N ≤ 20
3	7	K = 1
4	9	K = 2
5	15	주어지는 N개의 수가 단조증가(감소하지 않는 순서)로 정렬되어 있다. 이는 즉, N개의 수 각각에 대해 그 수의 왼쪽에는 해당 수 이하의 값들만 있고, 오른쪽에는 해당 수 이상의 값들만 있다는 뜻이다.
6	40	추가 제약 조건 없음

예제 입력 1 복사

5 3
2 3 1 2 1

예제 출력 1 복사

4

예제 입력 2 복사

6 2
4 1 5 2 6 3

예제 출력 2 복사

10

 

문제풀이(1)

n개의 수 중 k개를 고를 때, 전체 점수의 최댓값을 구하는 문제이다.
최댓값을 구하는 문제이기때문에 내림차순으로 정렬한뒤 큰 수 k개를 찾으면 될 것이다.
마지막으로 가장 큰 K개의 숫자의 합에서 K개의 숫자를 이루는 연속된 자연수의 합을 뺀 값을 구하면된다.

const filePath = process.platform === "linux" ? "/dev/stdin" : "./input.txt";
const input = require("fs").readFileSync(filePath).toString().trim().split("\n");

const [n, k] = input.shift().split(" ").map(Number);
const arr = input[0]
  .split(" ")
  .map(Number)
  .sort((a, b) => b - a);

let answer = 0;

for (let i = 0; i < k; i++) {
  answer += arr[i];
}

// 가장 큰 K개의 숫자의 합에서 K개의 숫자를 이루는 연속된 자연수의 합을 뺀 값
console.log(answer - (k * (k - 1)) / 2);