SGU 507 Treediff （dsu on tree）

题目链接：https://codeforces.com/problemsets/acmsguru/problem/99999/507

题意：n个点，组成一个树，根为1。这个树有m个叶子节点，每个叶子节点上都有一个值。对于n-m个非叶子节点，输出其子树上的叶子节点中，每对节点值的差的绝对值的最小值。

思路：有mp维护子树中叶节点的值，每插入一个值时，用第一个比他小的值和第一个比他大的值和他作差，更新答案。

代码：

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 5e4+10;
const ll inf = (1LL<<31)-1;
int n,m,val[N];
ll ans[N];
struct node{
	int to,nex;
}g[N];
int head[N],cnt;
void add(int u,int v){
	g[++cnt].to=v;
	g[cnt].nex=head[u];
	head[u]=cnt;
}
int sz[N],son[N],SON,limit,siz;
map<int,int> mp;
map<int,int>::iterator it;
void getsz(int u){
	sz[u]=1;
	int mx=-1;
	for(int i=head[u];i;i=g[i].nex){
		getsz(g[i].to);
		if(sz[g[i].to]>mx){
			mx=sz[g[i].to];
			son[u]=g[i].to;
		}
		sz[u]+=sz[g[i].to];	
	}
}
ll mn=inf;
void dfs1(int u){
	if(u>=limit){
		mp[val[u]]++;
		if(mp[val[u]]==1){
			it=mp.find(val[u]);	
			if(it!=mp.begin()){
				it--;
				mn=min(mn,0LL+val[u]-(*it).first);
				it++;
			}
			it++;
			if(it!=mp.end()) mn=min(mn,0LL+(*it).first-val[u]);
		}else mn=0;
	}
	for(int i=head[u];i;i=g[i].nex)
		if(g[i].to!=SON)
			dfs1(g[i].to);
			
}
void dfs(int u,bool keep){
	for(int i=head[u];i;i=g[i].nex){
		if(g[i].to!=son[u]){
			dfs(g[i].to,0);	
		}
	}
	if(son[u]){
		dfs(son[u],1);
		SON=son[u];
	}
	dfs1(u);
	SON=0;
	if(u<limit) ans[u]=mn;
	if(!keep){
		mp.clear();
		mn=inf;
	}
}
int main(void){
	scanf("%d%d",&n,&m);
	limit=n-m+1;
	ans[1]=inf;
	for(int i=2;i<=n;i++){
		int u;
		scanf("%d",&u);
		add(u,i);
		ans[i]=inf;
	}
	for(int i=limit;i<=n;i++)
		scanf("%d",&val[i]);
	getsz(1);
	dfs(1,0);
	
	printf("%lld",ans[1]);
	for(int i=2;i<limit;i++)
		printf(" %lld",ans[i]);
	return 0;	
}