【Ybtoj 第13章例1】合并果子【二叉堆】

---

知识点：关于堆

PS：还记得上周，我跟$aj$说我们没有完整学过堆，只是之前打过类似题目，然后$aj$大吃一惊，在机房问其他人，结果我震惊了，旁边的巨爷都说学过了，所以真的就我没打过堆（手打堆）吗？？？然后对面的$cyz$十分扎心地补了一句：“哦，那应该就。。（我）没有学吧”。。。然后aj就肯定的认为我之前没有来上课。。。我？？？
结果：我气的跳过了前面的章节，直接来学堆了。。

当然，我还是很懒不想总结，所以现在就到了安利博客的时间了：关于堆1+关于堆2

上面的博客，思路较清楚，但代码都不好看，所以，我来放一个很通俗的：

• put操作（向堆中添加新元素）

void put(int x){
	tree[++len]=x;
	int son=len;
	while(len>1)
	{
		int fa=son/2;
		if(tree[fa]<=tree[son])
		break;
		swap(tree[fa],tree[son]);
		son=fa;
	}
}

• get操作（取出堆顶元素）

int get(){
	int k=tree[1],fa=1;
	tree[1]=tree[len--];
	while(fa*2<=len)
	{
		int son=fa*2;
		if(son+1<=len&&tree[son+1]<tree[son])
			son+=1;
		if(tree[fa]<=tree[son])
			break;
		swap(tree[fa],tree[son]);
		fa=son;
	}
	return k;
}

---

解题思路

这是模板题，有个明显的贪心思路，每次合并重量最小的两堆，先把所有的数插入小根堆，每次取出当前最小的两个数，把他们的和计入答案并放回堆，循环$n-1$次

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;

int n,len,lyx,ans,a[10010],tree[10010];

void put(int x){
	tree[++len]=x;
	int son=len;
	while(len>1)
	{
		int fa=son/2;
		if(tree[fa]<=tree[son])
		break;
		swap(tree[fa],tree[son]);
		son=fa;
	}
}

int get(){
	int k=tree[1],fa=1;
	tree[1]=tree[len--];
	while(fa*2<=len)
	{
		int son=fa*2;
		if(son+1<=len&&tree[son+1]<tree[son])
			son+=1;
		if(tree[fa]<=tree[son])
			break;
		swap(tree[fa],tree[son]);
		fa=son;
	}
	return k;
}

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		put(a[i]);
	}
	for(int i=1;i<n;i++)
	{
		
		int kj=get(),lyx=get();
		ans+=kj+lyx;//把他们的和计入答案
		put(kj+lyx);//放回堆
	}
	printf("%d",ans);
}