Codeforces 1175D Array Splitting

傳送門

題意：給定一個數組a，將a分成k段，每一段都非空。其中如果一個數字a[i]在第j段中，那麼它的貢獻爲a[i]*j,現在需要最大化每個數字的貢獻之和。

思路：相當於在n-1個空中切k-1刀，設sum[i]爲i到n的後綴和。如果在a[i]之前切一刀，那麼貢獻值則會增加sum[i]，所以我們將後綴和排序，取最大的前k-1個即可。

 

#include <algorithm>
#include <iostream>
#include <cstdlib>
#include <cstring>
#include <cstdio>
#include <string>
#include <queue>
#include <cmath>
#include <stack>
#include <time.h>
#include <map>
#include <set>
#define mem(a,x) memset(a,x,sizeof(a))
#define gi(x) scanf("%d",&x)
#define gi2(x,y) scanf("%d%d",&x,&y)
#define gi3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define gll(x) scanf("%lld",&x)
#define gll2(x,y) scanf("%lld%lld",&x,&y)
using namespace std;
const double eps=1e-8; 
typedef long long ll;
const int MAXN=300005;
const ll mod=1e9+7;
const int inf=0x3f3f3f3f;
int n,k;
ll a[MAXN];
ll sum[MAXN];
ll ans=0;
bool cmp(ll a,ll b){
	return a>b;
}
int main(){
	int n,k;
	gi2(n,k);
	k--;
	for(int i=1;i<=n;i++){
		gll(a[i]);
		ans+=a[i];
	}
	for(int i=n;i>=1;i--){
		sum[i]=sum[i+1]+a[i];
	}
	for(int i=1;i<=n-1;i++){
		sum[i]=sum[i+1];
	}
	sort(sum+1,sum+n,cmp);
	for(int i=1;i<=k;i++){
		ans+=sum[i];
	}
	printf("%lld\n",ans);
	return 0;
}