优先队列priority_queue（重载）

最新推荐文章于 2023-10-28 23:25:09 发布

程序员的冷浪漫

最新推荐文章于 2023-10-28 23:25:09 发布

阅读量888

点赞数

分类专栏： 3、STL和c++学习

本文链接：https://blog.csdn.net/sunquana/article/details/10166717

版权

3、STL和c++学习专栏收录该内容

16 篇文章 1 订阅

订阅专栏

这个是算法可以求最小huffman树

#include <iostream>
#include <queue>
#include <vector>
#include <cstdio>
using namespace std;

struct Comp{
  bool operator()(const int &a,const int &b)
  {
      return a>b;
  }
};
int main()
{
    int i,n,num;
    while(cin>>n)
    {
        priority_queue<int,vector<int>,Comp>p;
        for(i=0;i<n;i++)
         {
             scanf("%d",&num);
             p.push(num);
         }
         if(n==1)
         { cout<<num<<endl;continue;}

         int temp,sum=0;
         for(i=1;i<n;i++)
         {
            temp=p.top(); p.pop();
            temp+=p.top(); p.pop();
             p.push(temp);
             sum+=temp;
            // cout<<temp<<" "<<"sum=="<<sum<<" ";
         }
         printf("%d\n",sum);
    }
    return 0;
}