【数据结构 | C++】数学家皮卡丘

现有元素个数为N的递增数组a。皮卡丘想把这个数组分成k个非空的连续子数组，使得数组中的每个元素都包含且仅包含在一个子数组中。设max(i)为第i个子数组中的最大元素，min(i)为第i个子数组中的最小元素。皮卡丘希望对于每个数组的[max(i)-min(i)]相加后取得最小值。

例如对于数组[2，4，5，5，8，11，19]，当给定k=3时应将其划分成3个子数组[2，4，5，5]，[8，11]，[19]，此时皮卡丘可以得到最小值为(5-2)+(11-8)+(19-19)=6 。

输入格式：第一行包括两个正整数n和k，分别表示a数组的元素个数和应划分出的子数组个数k(1≤k≤n≤3e5)。第二行输入N个正整数，分别表示数组元素ai。(0<ai<= 10^9)

输出格式：输出一个整数，为所求公式的最小值。

#include <bits/stdc++.h>
using namespace std;
int main(){
	int n,k;cin>>n>>k;int a[n],b[n-1];
	for(int i=0;i<n;i++)cin>>a[i];
	for(int i=0;i<n-1;i++)b[i]=a[i+1]-a[i];
	sort(b,b+n-1);int sum=0;//b[]用于存储两两差值
	for(int i=0;i<n-k;i++)sum+=b[i];cout<<sum;return 0;
}
//[4,8,15,16,23,42]，连续两个数的差分别为 4、7、1、7、19
//题目要求分为三组，则去除两个最大的差值19、7，得到结果4+7+1=12
//故要求分为K个数组时，只需去除K-1个最大的差值