#140 (div.1) B. Naughty Stone Piles

1.题目描述:点击打开链接

2.解题思路:本题可以转化为求解一棵k叉树的权值之和。根据题意过程类似于霍夫曼编码,但是由于每个点的合并次数最多只能是k,那么自然我们想到大的尽量少动,小的多移动,这样就可以想到构造一棵k叉树,从大到小依次排满根结点,第一层,第二层,每个结点的权值是重量乘以路径长度。这样便不难求解了。

3.代码:

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<algorithm>
#include<string>
#include<sstream>
#include<set>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<functional>
using namespace std;

typedef long long ll;
typedef unsigned long long ull;


#define me(s) memset(s,0,sizeof(s))
#define For(i,n) for(int i=0;i<(n);i++)
#define pb push_back
#define sz size
#define clr clear
#define F(a,b) for(int i=a;b;i++)


ll P[100005];
ll ans[100005];
int n;

int main()
{
    //freopen("t.txt","r",stdin);
    scanf("%d",&n);
    for(int i=0;i<n;i++)
        scanf("%I64d",&P[i]);
    sort(P,P+n);
    for(int i=1;i<n;i++)
        P[i]+=P[i-1];
    for(int i=1;i<=n;i++)
    {
        ll j=n-2,t=i;
        while(j>=0)
        {
            ans[i-1]+=P[j];
            j-=t;
            t*=i;
        }
    }
    int q;
    scanf("%d",&q);
    for(int i=0;i<q;i++)
    {
        int k;
        scanf("%d",&k);
        printf("%I64d ",k<n?ans[k-1]:ans[n-1]);
    }
    puts("");
    return 0;
}

你可能感兴趣的:(#140 (div.1) B. Naughty Stone Piles)