codeforces 1175D 后缀数组

题意：将数组分成连续的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;

}