Codeforces 1253/C. Sweets Eating（前缀和+找规律）

传送门

题意：给定一个长度为n的序列，代表每个糖果的糖度，每颗糖果经过d天后糖度将变为 d*ai ;
      xx每天最多可以吃k课糖果，问怎么选择糖果，使得吃掉 g 颗糖果xx获得的最少糖度；
      例如：
      n=9,k=2
      ai :   2   3   4   4   6   6   7   8   19 （排序后）
      输出： 2   5  11   18  30  43  62  83  121
      例如第g=5,选择：ai :2,3,4,4,6; 则第一天吃 4,6；第二天吃 3，4；第3天吃 2；
      这样获得的糖度为: (4+6)*1+(3+4)*2+2*3=30
分析：我们很容易知道，先吃糖度高的糖果这样才能保证总糖度最小；暴力数据肯定是不允许的，
     我们可以用前缀和来维护，再找规律。具体如下:
           i :   1   2   3   4   5   6   7   8   9
          ai :   2   3   4   4   6   6   7   8   19 （排序后）
前缀和  sumi :   2   5   9   13  19  25  32  40  59
  输出 ansi  :   2   5  11   18  30  43  62  83  121                
   k=2,   看 i=[3,5) : ans3=sum1+sum3; ans4=sum2+sum4;     
          看 i=[5,7) : ans5=sum1+sum3+sum5; ans6=sum2+sum4+sum6;
          看 i=[7,9) : ans7=sum1+sum3+sum5+sum7; ans8=sum2+sum4+sum6+sum8;
          ......     
          然后我们就可以用前缀和维护类似于这样每一个序列了：sum1+sum3+sum5+sum7...
                                                          ...

#include <set>
#include <map>
#include <cmath>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include<cstring>
#include <stdio.h>
#include <iostream>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn=1e6+5;
int t,n,k;
int a[maxn];
ll sum[maxn],ans[maxn],f[maxn];
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
        cin>>a[i];
    sort(a+1,a+n+1);
    sum[1]=a[1];
    for(int i=2;i<=n;i++)
        sum[i]=sum[i-1]+a[i];
    for(int i=1;i<=k;i++)
        ans[i]=sum[i];
    for(int i=1;i<=k;i++)
        f[i]=sum[i];
    for(int i=k+1;i<=n;i+=k){
        int l=1;
        while (l<=k){
            f[l]+=sum[i+l-1];
            ans[i+l-1]=f[l];
            l++;
        }
    }
    for(int i=1;i<=n;i++)
        printf("%lld ",ans[i]);
    printf("\n");
    return 0;
}