1-4 Huffuman树

目录

我的​代码​

网上的代码

学习反思

我的代码

#include<iostream>
using namespace std;

int cost(int* arr, int left, int right){
	int first = left, second = left + 1;
	if(right == second)
		return arr[first] + arr[second];
	if(arr[first] > arr[second] )
		swap(first,second);
	for(int i = left + 2; i <= right; i++)
		if (arr[i] < arr[second]){
			second = i;
			if(arr[first] > arr[second] )
				swap(first,second);
		}
	//找两个最小的数
	if(second == left){
		arr[first] += arr[second];
		return arr[first] + cost(arr, left+1, right);
	}
	else{
		swap(arr[first], arr[left]);
	//通过交换删除
		arr[second] += arr[left];
	//加入他们的和
		return arr[second] + cost(arr, left+1, right);
	}
}

int main(){
	int n, result;
	cin >> n;
	int *arr = new int[n];
	for (int i = 0; i < n; i++){
		cin >> arr[i];
	}
	result = cost(arr, 0, n-1);
	cout << result;
	return 0;
}

网上的代码

利用优先队列维护一个小顶堆,每次弹出两个最小值,然后将其之和再加入优先队列

#include<iostream>
#include<queue>
using namespace std;
int main()
{
	int n,sum=0;
	priority_queue<int ,vector<int>,greater<int> > q;
	cin >> n;
	while(n--)
	{
		int a;
		cin >> a;
		q.push(a);
	}
	while(q.size()!=1)
	{
		int a,b;
		a=q.top();
		q.pop();
		b=q.top();
		q.pop();
		sum+=a+b;
		q.push(a+b);
	}
	cout << sum;
	return 0;
}

学习反思

定义：priority_queue<Type, Container, Functional>
Type 就是数据类型，Container 就是容器类型（Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector），Functional 就是比较的方式，当需要用自定义的数据类型时才需要传入这三个参数，使用基本数据类型时，只需要传入数据类型，默认是大顶堆
一般是：

//升序队列
priority_queue <int,vector<int>,greater<int> > q;
//降序队列
priority_queue <int,vector<int>,less<int> >q;

//greater和less是std实现的两个仿函数（就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator()，这个类就有了类似函数的行为，就是一个仿函数类了）

————————————————
版权声明：上段为CSDN博主「吕白_」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/weixin_36888577/article/details/79937886