题目描述
总时间限制: 1000ms 内存限制: 65535kB
描述
构造一个具有n个外部节点的扩充二叉树,每个外部节点Ki有一个Wi对应,作为该外部节点的权。使得这个扩充二叉树的叶节点带权外部路径长度总和最小:
Min( W1 * L1 + W2 * L2 + W3 * L3 + … + Wn * Ln)
Wi:每个节点的权值。
Li:根节点到第i个外部叶子节点的距离。
编程计算最小外部路径长度总和。
输入
第一行输入一个整数t,代表测试数据的组数。
对于每组测试数据,第一行输入一个整数n,外部节点的个数。第二行输入n个整数,代表各个外部节点的权值。
2<=N<=100
输出
输出最小外部路径长度总和。
样例输入
2 3 1 2 3 4 1 1 3 5
样例输出
9 17
问题解决
使用优先队列,使用方法如下:
//小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//大顶堆
priority_queue <int,vector<int>,less<int> >q;
每次取两个最小的节点,和加上两节点值之和,再将其合并加入队列,直到队列中只剩下一个节点为止:
#include <iostream>
#include <queue>
using namespace std;
int main(){
int N,n,tmp;
cin>>N;
while(N--){
cin>>n;
priority_queue<int,vector<int>,greater<int> > q;
int res=0,x,y;
for(int i=0;i<n;i++){
cin>>tmp;
q.push(tmp);
}
while(q.size()>1){
x=q.top();q.pop();
y=q.top();q.pop();
res+=(x+y);
q.push(x+y);
}
cout<<res<<endl;
}
return 0;
}