Treediff SGU-507树上的启发式合并

题目链接：SGU-507

题目思路：

（启发式合并基本套路）每次合并时将子树节点少的合到子树节点多的，合并时更新答案，同时更新当前子树的根节点的答案。

#include<cstdio>
#include<algorithm>
#include<set>
#define INF 2147483647
#define M 50005
using namespace std;
struct E {//正向表存边 
	int to,nx;
} edge[M];
int val[M];
int tot,head[M];
void Addedge(int a,int b) {
	edge[++tot].to=b;
	edge[tot].nx=head[a];
	head[a]=tot;
}
set<int>S[M];
set<int>::iterator it1,it2;
int ans[M];
int merge(int a,int b) {
	if(S[a].size()<S[b].size())swap(S[a],S[b]);
	int res=INF;//res为当前子树的答案 
	for(it1=S[b].begin(); it1!=S[b].end(); it1++) {
		it2=S[a].lower_bound(*it1);//找离他的值最近的两个值 
		if(it2==S[a].begin()) {
			res=min(res,*it2-*it1);
		} else if(it2==S[a].end()) {
			it2--;
			res=min(res,*it1-*it2);
		} else {
			res=min(res,*it2-*it1);
			it2--;
			res=min(res,*it1-*it2);
		}
		S[a].insert(*it1);//这样影响答案，因为若有更优解仍会更新 
	}
	S[b].clear();
	return res;
}
void dfs(int now) {
	ans[now]=INF;
	if(!head[now]) {//是叶子结点 
		S[now].insert(val[now]);
		return;
	}
	for(int i=head[now]; i; i=edge[i].nx) {
		int nxt=edge[i].to;
		dfs(nxt);
		ans[now]=min(ans[now],ans[nxt]);
		ans[now]=min(ans[now],merge(now,nxt));//一边更新一边合并 
	}
}
int main() {
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=2; i<=n; i++) {
		int x;
		scanf("%d",&x);
		Addedge(x,i);
	}
	for(int i=n-m+1; i<=n; i++)scanf("%d",&val[i]);
	dfs(1);
	for(int i=1; i<=n-m; i++)printf("%d ",ans[i]);
}