Treediff

一、题目

题目描述

$n$个点的树，有$m$个叶子结点，只有叶子结点有权值，问每一个非叶节点的子树内选两个不同的叶子的最小差，如果不存在输出$2^{31}-1$

数据范围

$1\le m<n\le 50000$

二、解法

启发式合并的版题，随便取一个子树的$set$，然后启发式合并，把小的并到大的那里去，随便算答案就行了。

证明一下这样做的复杂度，每次合并后的数量一定是小的集合大小的两倍，最开始的大小都为$1$，合并$t$次后大小就是$2^t$，故每个点只会合并$t=\log$次，总复杂度就是$O(n{\log }^{2}n)$

#include <cstdio>
#include <iostream>
#include <set>
using namespace std;
const int M = 50005;
int read()
{
    int num=0,flag=1;char c;
    while((c=getchar())<'0'||c>'9')if(c=='-')flag=-1;
    while(c>='0'&&c<='9')num=(num<<3)+(num<<1)+(c^48),c=getchar();
    return num*flag;
}
int n,m,tot,f[M],b[M],res[M];
set<int> a[M];
struct edge
{
	int v,next;
}e[2*M];
int dfs(int u,int fa)
{
	int ans=2147483647;
	for(int i=f[u];i;i=e[i].next)
	{
		int v=e[i].v;
		if(v==fa) continue;
		ans=min(ans,dfs(v,u));
		if(!b[u]) {b[u]=b[v];continue;}
		int len1=a[b[u]].size(),len2=a[b[v]].size();
		set<int>::iterator it,now;
		if(len1>len2)
		{
			for(it=a[b[v]].begin();it!=a[b[v]].end();it++)
			{
				now=a[b[u]].lower_bound(*it);
				if(now!=a[b[u]].end()) ans=min(ans,(*now)-(*it));
				if(now!=a[b[u]].begin()) now--,ans=min(ans,(*it)-(*now));
				a[b[u]].insert(*it);
			}
		}
		else
		{
			for(it=a[b[u]].begin();it!=a[b[u]].end();it++)
			{
				now=a[b[v]].lower_bound(*it);
				if(now!=a[b[v]].end()) ans=min(ans,(*now)-(*it));
				if(now!=a[b[v]].begin()) now--,ans=min(ans,(*it)-(*now));
				a[b[v]].insert(*it);
			}
			b[u]=b[v];
		}
	}
	return res[u]=ans;
}
int main()
{
	n=read();m=read();
	for(int i=2;i<=n;i++)
	{
		int j=read();
		e[++tot]=edge{i,f[j]},f[j]=tot;
		e[++tot]=edge{j,f[i]},f[i]=tot;
	}
	for(int i=n-m+1;i<=n;i++)
		a[i].insert(read()),b[i]=i;
	dfs(1,0);
	for(int i=1;i<=n-m;i++)
		printf("%d ",res[i]);
}