OJ_哈夫曼树

frostmelody

于 2024-03-09 15:37:31 发布

阅读量217

点赞数 2

CC 4.0 BY-SA版权

分类专栏：数据结构与算法文章标签： c++ 算法

本文链接：https://blog.youkuaiyun.com/Listennnn/article/details/136584467

数据结构与算法专栏收录该内容

91 篇文章

订阅专栏

该代码展示了如何使用C++中的优先队列实现小根堆，通过不断选择并合并两个最小元素，直至只剩一个元素，最后输出其相反数作为结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题干

在这里插入图片描述

C++实现

从小根堆的选择最小的两个
合并加和，然后插入回小根堆
然后重复上述两个步骤，直到只剩一个元素

#define _CRT_SECURE_NO_WARNINGS
#include <stdio.h>
#include <queue>
#include <string>
using namespace std;


int main() {
	int n;
	scanf("%d", &n);
	priority_queue<int> pqueue;//存储权值相反数,以实现小根堆效果
	for (int i = 0; i < n; i++)
	{
		int leaf;
		scanf("%d", &leaf);
		pqueue.push(-1*leaf);
	}
	int wpl = 0;
	while (pqueue.size() >= 2) {
		int leaf1 = pqueue.top();
		pqueue.pop();
		int leaf2 = pqueue.top();
		pqueue.pop();
		wpl += leaf1 + leaf2;
		pqueue.push(leaf1 + leaf2);
	}
	printf("%d\n", -1 * wpl);
}