题意
给你一个序列,要求你将其分割为k个子序列,第一个序列权值为1,第二个权值为2,第n个权值为n 问k分别为1-n时,该序列的权值最大为多少?
思路
其他的题解基本上都是求后缀和排序来解的,由于我们队赛时用求前缀和的方式AC,这里分享一个前缀和的做法,本质上和求后缀和是一样的。
由于第一段权值为1,第二段权值为2,我们可以将其想象为n条线段,每一条被割去了一部分,即割去了一个前缀
红色部分为被割去部分,很显然越早割去的前缀越小越好
所以,我们可以求出前缀和后进行一次排序,将前缀从小到大依次割去(不包括最后一个前缀数,因为最后一个前缀代表整条线段)
代码
int a[N],pre[N];
void solve() {
int n=read();
for(int i=1;i<=n;i++){
a[i]=read(),pre[i]=pre[i-1]+a[i];
}
sort(pre+1,pre+n);//对前缀进行排序
for(int i=1;i<n;i++){//将前n-1个前缀进行累加,便于计算
pre[i]+=pre[i-1];
}
for(int i=1;i<=n;i++){
cout<<pre[n]*i-pre[i-1]<<' ';//线段的总长度-割去部分长度
}
cout<<endl;
}