CF1175D 思维题 后缀和数组

数组分成K段，第K段的数会被累加k次，所以用后缀和数组找到累加值最大的k-1段，再加上整个数组的和就是答案。

#include<iostream>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<cstdio>
using namespace std;
typedef long long ll;
int a[300010];
ll sum[300010];
int main()
{
	int n,k;
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	int cnt=0;
	for(int i=n;i>0;i--)
	{
		cnt++;
		sum[cnt]+=a[i]+sum[cnt-1];		
	}	
	ll ans=0;
	sum[cnt]+=1e15;
	sort(sum+1,sum+1+cnt);
	for(int i=cnt;i>=cnt-k+1;i--)
		ans+=sum[i];
	ans-=1e15;
	cout<<ans<<endl;	
}