这道题要用哈夫曼树,不过k=3或2。一开始想到补0,需要补的个数为k-(n-1)%(k-1)-1。但后来发现,哈夫曼树≠平衡树,n=3的时候都会出现问题。然后多试了几个例子(n=2,3,4,5,6,7,8,9),发现当n是偶数的时候,需要将最小的两个数相加(k=2),然后剩下的情况都是k=3;而n是奇数的时候,全部k=3就可以。
用优先权队列来做:
#include<queue>
using namespace std;
priority_queue<int,vector<int>,less<int> >que; //从大到小
priority_queue<int,vector<int>,greater<int> >que; //从小到大
附上AC代码:
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
#define ll long long
const int INF=0x3f3f3f3f;
//priority_queue<int,vector<int>,less<int> >que; //从大到小
priority_queue<ll,vector<ll>,greater<ll> >que; //从小到大
///注意long long类型!
int main()
{
int n;
while(scanf("%d",&n)==1)
{
while(!que.empty()) que.pop();//清空队列
ll x;
for(int i=0;i<n;i++)
{
scanf("%lld",&x);
que.push(x);
}
if(n==1)
{
printf("0\n");
continue;
}
ll ans=0;
if(que.size()%2==0)//n是偶数,第一步两个数相加
{
ll t=que.top(); que.pop();
t+=que.top(); que.pop();
ans+=t;
que.push(t);
}
while(que.size()>1)
{
ll t=que.top(); que.pop();
t+=que.top(); que.pop();
t+=que.top(); que.pop();
ans+=t;
que.push(t);
}
printf("%lld\n",ans);
}
return 0;
}
/**
6
1 4 4 4 4 4
ans=38
7
1 4 4 4 4 4 4
ans=46
**/