C. Array Splitting
time limit per test2 seconds
memory limit per test256 megabytes
inputstandard input
outputstandard output
You are given a sorted array a1,a2,…,an (for each index i>1 condition ai≥ai−1 holds) and an integer k.
You are asked to divide this array into k non-empty consecutive subarrays. Every element in the array should be included in exactly one subarray.
Let max(i) be equal to the maximum in the i-th subarray, and min(i) be equal to the minimum in the i-th subarray. The cost of division is equal to ∑i=1k(max(i)−min(i)). For example, if a=[2,4,5,5,8,11,19] and we divide it into 3 subarrays in the following way: [2,4],[5,5],[8,11,19], then the cost of division is equal to (4−2)+(5−5)+(19−8)=13.
Calculate the minimum cost you can obtain by dividing the array a into k non-empty consecutive subarrays.
Input
The first line contains two integers n and k (1≤k≤n≤3⋅105).
The second line contains n integers a1,a2,…,an (1≤ai≤109, ai≥ai−1).
Output
Print the minimum cost you can obtain by dividing the array a into k nonempty consecutive subarrays.
Examples
inputCopy
6 3
4 8 15 16 23 42
outputCopy
12
inputCopy
4 4
1 3 3 7
outputCopy
0
inputCopy
8 1
1 1 2 3 5 8 13 21
outputCopy
20
Note
In the first test we can divide array a in the following way: [4,8,15,16],[23],[42].
思想:
题意:n个数选区k个区间求每个区间的最后一个数减去第一个数 然后将得到的值的和最小
根据题意有
a[k1]-a[1]+a[k2]-a[k1+1]+…+an-a[k(n-1)+1]=a[n]-a[1]-(a[k1+1)-a[k1]]+…+a[k(n-1)+1]-a[k(n-1)];
可以看出要使上式结果最小,要减去除去前两个元素k-1个最大的相邻元素的差
可以维护一个优先队列
include<stdio.h>
#include<queue>
int a,b,c,n,k;std::priority_queue<int>Q;
int main(){
scanf("%d%d%d",&n,&k,&b);a=c=b;
for(int i=2;i<=n;i++,b=a){
scanf("%d",&a);
Q.push(a-b);
}c=a-c;
while(--k&&!Q.empty()){
c-=Q.top();Q.pop();
}printf("%d",c);
}