题意:将数组分成连续的k个集合,然后定义花费为 每个集合的和*集合的下标求最大花费
题解:对应的花费就是后缀数组选出k个来加和(画画图看看),其中sum[1]必须包括,剩下的就是选出k-1个后缀数组来加和就是对应的花费了
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+10, INF = 0x3f3f3f3f;
#define pi acos(-1.0)
int a[maxn];ll sum[maxn];
int main() {
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
for(int i=n;i>0;i--){
sum[i]=a[i]+sum[i+1];
}
sort(sum+2,sum+1+n);
ll ans=sum[1];
for(int i=0;i<k-1;i++){
ans+=1ll*sum[n-i];
}
cout<<ans<<endl;
}