树-堆结构练习——合并果子之哈夫曼树

来源:互联网 发布:sql美化工具 编辑:程序博客网 时间:2024/05/16 16:08

Description

 在一个果园里,多多已经将所有的果子打了下来,而且按果子的不同种类分成了不同的堆。多多决定把所有的果子合成一堆。
每一次合并,多多可以把两堆果子合并到一起,消耗的体力等于两堆果子的重量之和。可以看出,所有的果子经过n-1次合并之后,就只剩下一堆了。多多在合并果子时总共消耗的体力等于每次合并所消耗体力之和。

Input

 第一行是一个整数n(1<=n<=10000),表示果子的种类数。第二行包含n个整数,用空格分隔,第i个ai(1<=ai<=20000)是第i个果子的数目。
 

Output

 输出包括一行,这一行只包含一个整数,也就是最小的体力耗费值。输入数据保证这个值小于2^31。
 

Sample Input

31 2 9

Sample Output

15
思路:每次都将最小的两个相加,比如1 2 2 9,先将1和2相加为3,变成2,3,9,再将2和3相加为5,最后成5,9相加为14,最后再将每次的花费相加,不要理解错题意了╮(╯▽╰)╭。。。
过程:用的优先队列,priority_queue<int,vector<int>,greater<int> >q;将队列从小到大自动排列,想将栈顶加入,弹出,在加入下一个栈顶,弹出,将和加入队列,用一个总和来记录每次和的总和,一共循环n-1次直到结束\(≧▽≦)/
代码实现:
#include<stdio.h>#include<vector>#include<queue>#include<iostream>#include <algorithm>using namespace std;queue<int>q;int main(){    int n,m,i,j;    priority_queue<int,vector<int>,greater<int> >q;    cin>>n;    for(i=0; i<n; i++)    {        cin>>m;        q.push(m);    }    int sum,cou=0;    int st;    n=n-1;    while(n--)    {        sum=0;        st=q.top();        sum+=st;        q.pop();        st=q.top();        sum+=st;        q.pop();        cou+=sum;        q.push(sum);       priority_queue<int,vector<int>,greater<int> >q;    }    printf("%d\n",cou);    return 0;}


 

0 0
原创粉丝点击