cf 226b 唯美思维题~.cpp

题意：

　　给出n个数，让你用最少次数把所有的数都摞成一堆。

　　其中当把arr[i]摞在第j堆上时，得扣分arr[i]

　　题目有要求每个数上面不许摞超过k次，有m个k，让你求出m种情况下限制为k时的最优解，即最少扣分。

思路：

　　经过多次推算会发现，如果要最后扣分最少，则应该尽量让arr[i]小的摞的次数比较多，而arr[i]大的摞的次数比较少。

　　然后还有一个规律是，如果想让扣分最少，那每次应该尽量让值大的堆上摞的次数比较多，本身则尽量少摞。即尽量满足 一次摞足够多的物品。

Tips：

　　如果k = 1，则答案就是arr[1] + (arr[1]+arr[2]) + (arr[1]+arr[2]+arr[3]) + (arr[1] +...+arr[i]) + (arr[1]+...+arr[i]+...+arr[n-1])//因为最后且最大的一摞不用动~所以只需要加到arr[n-1]

　　否则答案就是一次摞尽量多的物品，为了令大的物品移动尽量少的次数，我们从倒数第二个到倒数第k+1个开始算，arr[n-1]~arr[n-1-k]只需要移动一次，arr[n-1-k]~arr[n-1-k-k*k]只要移动2次..and so on~~//因为1个上面可以摞k个..所以k个上面可以摞k*k个..所以以倍数的方式递增..

　　优化的方法是先计算出从第0个到第i个的和，这样就可以算出其中任意一段的和了。

　　所以答案就是while循环算出来的..

Code：

#include <stdio.h>
#include <cstring>
#include <algorithm>
#include <ctime>
using namespace std;

long long sum[100010];
long long arr[100010];

int main()
{
    int n, q, k;
    long long ans, tans, tk;
    int tmp, tn;
    while(EOF != scanf("%d", &n)) {
        tans = 0;
        for(int i = 1; i <= n; ++i)
            scanf("%lld", &arr[i]);
        //printf("_\n");
        sort(arr+1, arr+n+1);
        sum[0] = 0;
        for(int i = 1; i <= n; ++i) {
            sum[i] = sum[i-1]+arr[i];
            if(i != n) tans += sum[i];
        }
        scanf("%d", &q);
        while(q--) {
            ans = 0;
            tmp = 1;
            tn = n-1;
            scanf("%d", &k);
         //   printf("%d__\n",k);
            tk = k;
            if(k != 1) {
                while(tn >= tk) {
                    ans += (sum[tn]-sum[tn-tk])*tmp;
                    tn -= tk;
                    tk *= k;
                    tmp++;
                }
                ans += (sum[tn]-sum[0])*tmp;
            } else ans = tans;
            printf("%lld%c", ans, q != 0?' ':'\n');
        }
    }
    return 0;
}

链接：http://www.codeforces.com/problemset/problem/226/B