设 第 i 组 在 p i 点 后 结 束 , s u m n 数 组 是 前 缀 和 数 组 , 那 么 可 得 答 案 是 设第i组在p_i点后结束,sumn数组是前缀和数组,那么可得答案是 设第i组在pi点后结束,sumn数组是前缀和数组,那么可得答案是

a n s = ∑ i = 1 i = k ( s u m n i − s u m n i − 1 ) ∗ i ans=\sum_{i=1}^{i=k}(sumn_i-sumn_{i-1})*i ans=∑i=1i=k(sumni−sumni−1)∗i

这 个 式 子 相 当 于 g x 1 + g x 2 + g x 3 + . . . g x k 这个式子相当于g_{x_1}+g_{x_2}+g_{x_3}+...g_{x_k} 这个式子相当于gx1+gx2+gx3+...gxk

其 中 g 表 示 后 缀 和 其中g表示后缀和 其中g表示后缀和

所 以 把 除 [ 1 , n ] 的 所 有 后 缀 和 排 个 序 选 k − 1 个 , 然 后 [ 1 , n ] 是 必 选 的 所以把除[1,n]的所有后缀和排个序选k-1个,然后[1,n]是必选的 所以把除[1,n]的所有后缀和排个序选k−1个,然后[1,n]是必选的

因 为 选 了 这 k − 1 个 后 缀 可 以 完 成 分 成 k 组 , 且 此 时 的 和 就 是 答 案 , 排 序 后 是 最 大 的 因为选了这k-1个后缀可以完成分成k组,且此时的和就是答案,排序后是最大的 因为选了这k−1个后缀可以完成分成k组,且此时的和就是答案,排序后是最大的

​​这篇博客有一张图可以帮助理解​​

#include <bits/stdc++.h>
using namespace std;
#define int long long
const int maxn=3e5+10;
int n,m,a[maxn],ans,pre[maxn],k;
bool com(int a,int b){
  return a>b;
}
signed main()
{
  cin >> n >> k;
  for(int i=1;i<=n;i++) cin >> a[i];
  //*1,*2,*3,*4......
  //1 1 -99999999,1 2 3 4 5 
  for(int i=n;i>=1;i--)
    pre[i]=pre[i+1]+a[i];
  sort(pre+2,pre+1+n,com);
  for(int i=1;i<=k;i++) ans+=pre[i];
  cout << ans;
}