D. Array Splitting（贪心）

最新推荐文章于 2024-06-16 13:34:59 发布

原创最新推荐文章于 2024-06-16 13:34:59 发布 · 327 阅读

0 ·

CC 4.0 BY-SA版权

aa题目题解专栏收录该内容

408 篇文章

订阅专栏

本文介绍了一种解决特定数组分割问题的算法思路，通过计算数字1到n的后缀和，找出并累加后缀中前（k-2）个最大值，以此达到优化求和结果的目的。该算法首先计算后缀和，然后对这些和进行排序，最终求得最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

D. Array Splitting

思路：

求出数字1~n的后缀和，除去sum[1~n],求出后缀中前（k-2）个最大值，（这样就使较大的值累加了k-2次），

然后结果就是sum[1~n]+ans。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn = 3e5+10;
LL a[maxn]={0},re[maxn]={0};
int main(void){
	int n,k;scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
	for(int i=n;i>=1;i--) re[i] = re[i+1]+a[i];
	sort(re+2,re+n+1);
	LL ans = re[1];
	for(int i=n;i>=n-k+2;i--) ans += re[i];
	printf("%lld\n",ans);
	
	return 0;
}