SGU 507 Treediff (线段树合并)

求每个子树的叶子节点差值的最小值，合并时用右子树的最小值和左子树的最大值的差更新答案

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
vector<int>ve;
const int MAXN=(int)5e4+10;
const int INF =INT_MAX;
int root[MAXN],tot,ls[MAXN*17],rs[MAXN*17],sz[MAXN*17];
int d[MAXN*17],vl[MAXN],ma[MAXN*17],mi[MAXN*17];
void insert(int l,int r,int x,int &rt){
	if(rt==0)rt=++tot;		
	ma[rt]=mi[rt]=x;sz[rt]=1;d[rt]=INF;
	if(l==r)return ;
	int mid=l+r>>1;
	if(mid>=x)insert(l,mid,x,ls[rt]);
	else insert(mid+1,r,x,rs[rt]);
}
int merge(int u,int v){
	if(u==0||v==0)return u|v;
	ls[u]=merge(ls[u],ls[v]);
	rs[u]=merge(rs[u],rs[v]);
	sz[u]+=sz[v];
	mi[u]=min(mi[u],mi[v]);
	ma[u]=max(ma[u],ma[v]);
	d[u]=min(d[u],d[v]);
	if(ls[u]!=0&&rs[u]!=0)d[u]=min(d[u],ve[mi[rs[u]]-1]-ve[ma[ls[u]]-1]);
	if(ls[u]!=0)d[u]=min(d[u],d[ls[u]]);
	if(rs[u]!=0)d[u]=min(d[u],d[rs[u]]);
	if(ls[u]==0&&rs[u]==0&&sz[u]>1)d[u]=0;
	return u;
}
struct edge{int to,nxt;}ed[MAXN<<1];
int head[MAXN],cnt;
int ans[MAXN];
void addedge(int u,int v){
	ed[cnt].to=v;
	ed[cnt].nxt=head[u];
	head[u]=cnt++;
}
int n,m,vsz;
void dfs(int u){
	for(int i=head[u];i!=-1;i=ed[i].nxt){
		int v=ed[i].to;
		dfs(v);
		root[u]=merge(root[u],root[v]);
	}	
	if(u+m>n)insert(1,vsz,vl[u],root[u]);
	else ans[u]=d[root[u]];
}
int main()
{
	memset(head,-1,sizeof(head));
	scanf("%d%d",&n,&m);
	for(int i=2;i<=n;i++){
		int x;
		scanf("%d",&x);
		addedge(x,i);
	}
	for(int i=1;i<=m;i++){
		scanf("%I64d",&vl[n-m+i]);
		ve.push_back(vl[n-m+i]);
	}
	sort(ve.begin(),ve.end());
	ve.erase(unique(ve.begin(),ve.end()),ve.end());
	vsz=ve.size();
	for(int i=n-m+1;i<=n;i++)vl[i]=lower_bound(ve.begin(),ve.end(),vl[i])-ve.begin()+1;
	dfs(1);	
	for(int i=1;i<=n-m;i++)printf("%d ",ans[i]);
	printf("\n");
	return 0;
}