题目大意
给定T组数据
有一个长度为n 的正整数数组A,满足Ai >= Ai+1,现在构造一个数组B,令B=A的后缀和。
现在,有一个n * n 的网格图,左下角坐标是(1, 1),右上角坐标是(n, n)。有一个小SB正在坐标为(n, 1) 的位置,每一时刻,如果他现在在(x, y),他可以选择走到(x ?-1,y + 1) 或者(x, (y + 1) div 2),如果选择后者,他要支付Bx的代价。
现在他想走到(1, 1),你可以告诉他他支付的代价最少是多少吗?注意在任何时候他都不能离开这个网格图。
题目解析
这个问题的答案就是数组 A 中所有元素的哈夫曼树的权值。
就用优先队列来解决,每次取优先队列中最小的两个数,即队列的第一个和第二个合并后再加入队列中
原因:感性理解一下吧
其实就是与合并果子(2004年NOIP提高组)几乎一样
代码
#include<bits/stdc++.h>
#define LL long long
using namespace std;
priority_queue<LL,vector<LL>,greater<LL> > q;
LL n,a[100001],t1,t2,ans,T;
int main()
{
cin>>T;
while(T--)
{
ans=0;
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
q.push(a[i]);
}
for(int i=1;i<n;i++)
{
t1=q.top();
q.pop();
t2=q.top();
q.pop();
ans+=t1+t2;
q.push(t1+t2);
}
cout<<ans<<endl;
while(!q.empty()) q.pop();
}
}