Codeforces 1175D. Array Splitting
传送门:https://codeforces.com/problemset/problem/1175/D
提醒自己:
要注意运算过程中的常数属性。比如这里面accumulate中的0一定为ll型。
题目大意:
题目会给你两个数字n,k以及一个长度为n的数组,你的任务是把这个数组分成连续的n段,使得第i段的数字总和乘上i之后,它们的总和最大。
主要想法:
本来我的想法是,一个数如果分到的数组组数越大,它的作用也就越大,所以,我们应该让大的数尽量在后面的组里,你让在数加入后面的组后,不可避免地也会加入负数。所以我们应该算的是从后往前的数的和来确定分组。但是如果把每个数当作单独的整体话,就会使得每次取走一些数后,就得重新求和,这样肯定T。所以,应该换个思路,因为每个第k组数出现k次,那么我从中取k-1次给k-1组,它自己还剩1次,依此类推,则每一组都变成了自己的数字加上后面所有的数字,第一组变成了整个数列。根据这种思想,题意可以转化为有n个数最后一个数是数组最后一个数,它前面一个数为它加上数组倒数第二个数,依此类推。最后只要先出除了第一个数之外最大的k-1个数即可。
代码如下:
#include <bits/stdc++.h>
using namespace std;
int main()
{
long long n,k,d=0;
cin >>n>>k;
vector<long long> a(n),b(n);
for(int i=n-1;i>=0;--i)
{
cin >>a[i];
}
b[0]=a[0];
for(int j=1;j<n;++j)
{
b[j]=a[j]+b[j-1];
}
sort(b.begin(),b.end()-1);
d=accumulate(b.begin()+b.size()-k,b.end(),(long long)0);
cout <<d<<endl;
return 0;
}