CodeForces 313 C.Ilya and Matrix（构造）

Description

给出$4^n$个数，要求构造一个$2^n\times 2^n$的矩阵，使得该矩阵的权值最大，定义一个$2^n\times 2^n$矩阵的权值：令$m$为该矩阵中元素的最大值，如果$n=0$，则该矩阵的权值为$m$;否则把该矩阵分成四个$2^{n-1}\times 2^{n-1}$的子矩阵，该矩阵的权值为这四个四矩阵的权值和加上$m$

Input

第一行输入一个整数$4^n$，之后输入$4^n$个整数$a_{i}$ $(1\leq 4^{n}\leq 2\times 10^6,1\leq a_{i}\leq 10^9)$

Output

输出由这$4^n$个数字构成的$2^n\times 2^n$的矩阵的最大权值

Sample Input

4

1 2 3 4

Sample Output

14

Solution

通过适当的构造可以使得在递归求权值时，第一次取到最大值，第二次取到最大的4个值，第三次取到最大的$4^2$个值，……，第$n$次取到所有的$4^n$个值，对$a_{i}$降序排后求个前缀和，$\sum\limits_{i=1}^na[4^i]$即为答案

Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 2222222
int n;
ll a[maxn];
bool cmp(ll a,ll b)
{
    return a>b;
}
int main()
{
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)scanf("%I64d",&a[i]);
        sort(a+1,a+n+1,cmp);
        for(int i=2;i<=n;i++)a[i]+=a[i-1];
        ll ans=0;
        for(int i=1;i<=n;i*=4)ans+=a[i];
        printf("%I64d\n",ans);
    }
    return 0;
}