【BZOJ4317】Atm的树

Description

Atm有一段时间在虐qtree的题目，于是，他满脑子都是tree，tree，tree……

于是，一天晚上他梦到自己被关在了一个有根树中，每条路径都有边权，一个神秘的声音告诉他，每个点到其他的点有一个距离（什么是距离不用说吧），他需要对于每个点回答：从这个点出发的第k小距离是多少；

如果atm不能回答出来，那么明天4019的闹钟将不会响，4019全寝可能就迟到了，所以atm希望你帮帮他。

Input

第一行，两个正整数n，k，表示树的点数，询问的是第几小距离；

第二~n行，每行三个正整数x，y，w，表示x和y之间有一条边，x为父亲，边权为w；

Output

n行，每行一个数，第i行输出从i开始第k小距离

Sample Input

5 2
1 5 2
1 2 4
2 3 6
2 4 5

Sample Output

4
5
10
9
6

HINT

100% n<=15000, 边权在1~10之间，为了方便，保证1为根；K<=5000

题解：依旧是动态点分治。

统计第k大不太好搞，我们对于每个点都二分一下，变成求到一个点距离<=mid的点有多少个，然后就变成熟悉的题了。

我们对于每个点维护一棵线段树，记录它在点分树的子树中有多少个点到它的距离为x，同时为了去重，还要维护一个从它父亲中减去的版本。

时间复杂度$O(nlog_n^3)$。

#include <cstdio>

#include <cstring>

#include <iostream>

using namespace std;

const int maxn=15010;

int n,m,N,tot,cnt,rt,mn;

int to[maxn<<1],next[maxn<<1],head[maxn],val[maxn<<1],siz[maxn],fa[maxn],md[20][maxn<<1],pos[maxn],vis[maxn];

int Log[maxn<<1],dep[maxn],r1[maxn],r2[maxn];

struct sag

{

	int ls,rs,siz;

}s[maxn*100];

inline int rd()

{

	int ret=0,f=1;	char gc=getchar();

	while(gc<'0'||gc>'9')	{if(gc=='-')	f=-f;	gc=getchar();}

	while(gc>='0'&&gc<='9')	ret=ret*10+gc-'0',gc=getchar();

	return ret*f;

}

inline void add(int a,int b,int c)

{

	to[cnt]=b,val[cnt]=c,next[cnt]=head[a],head[a]=cnt++;

}

void getrt(int x,int fa)

{

	int tmp=0,i;

	siz[x]=1;

	for(i=head[x];i!=-1;i=next[i])	if(!vis[to[i]]&&to[i]!=fa)

		getrt(to[i],x),tmp=max(tmp,siz[to[i]]),siz[x]+=siz[to[i]];

	tmp=max(tmp,tot-siz[x]);

	if(tmp<mn)	mn=tmp,rt=x;

}

void dfs(int x)

{

	pos[x]=++pos[0],md[0][pos[0]]=dep[x];

	for(int i=head[x];i!=-1;i=next[i])	if(!dep[to[i]])	dep[to[i]]=dep[x]+val[i],dfs(to[i]),md[0][++pos[0]]=dep[x];

}

void solve(int x)

{

	vis[x]=1;

	for(int i=head[x];i!=-1;i=next[i])	if(!vis[to[i]])

		tot=siz[to[i]],mn=1<<30,getrt(to[i],x),fa[rt]=x,solve(rt);

}

inline int dis(int x,int y)

{

	int a=pos[x],b=pos[y];

	if(a>b)	swap(a,b);

	int k=Log[b-a+1];

	return dep[x]+dep[y]-2*min(md[k][a],md[k][b-(1<<k)+1]);

}

void insert(int l,int r,int &x,int a)

{

	if(!x)	x=++tot;

	s[x].siz++;

	if(l==r)	return ;

	int mid=(l+r)>>1;

	if(a<=mid)	insert(l,mid,s[x].ls,a);

	else	insert(mid+1,r,s[x].rs,a);

}

int query(int l,int r,int x,int a,int b)

{

	if(a>b)	return 0;

	if(!x||(a<=l&&r<=b))	return s[x].siz;

	int mid=(l+r)>>1;

	if(b<=mid)	return query(l,mid,s[x].ls,a,b);

	if(a>mid)	return query(mid+1,r,s[x].rs,a,b);

	return query(l,mid,s[x].ls,a,b)+query(mid+1,r,s[x].rs,a,b);

}

inline int calc(int x,int mid)

{

	int ret=0,y,z;

	for(y=x;y;y=z)

	{

		z=fa[y];

		if(z)	ret-=query(0,N,r2[y],0,mid-dis(x,z));

		ret+=query(0,N,r1[y],0,mid-dis(x,y));

	}

	return ret;

}

int main()

{

	n=rd(),m=rd()+1;

	int i,j,u,v,a,b,c,l,r,mid;

	memset(head,-1,sizeof(head));

	for(i=1;i<n;i++)	a=rd(),b=rd(),c=rd(),add(a,b,c),add(b,a,c),N+=c;

	dep[1]=1,dfs(1),tot=n,mn=1<<30,getrt(1,0),solve(rt);

	for(i=2;i<(n<<1);i++)	Log[i]=Log[i>>1]+1;

	for(j=1;(1<<j)<(n<<1);j++)	for(i=1;i+(1<<j)-1<(n<<1);i++)	md[j][i]=min(md[j-1][i],md[j-1][i+(1<<(j-1))]);

	for(i=1;i<=n;i++)

	{

		for(u=i;u;u=v)

		{

			v=fa[u];

			if(v)	insert(0,N,r2[u],dis(i,v));

			insert(0,N,r1[u],dis(i,u));

		}

	}

	for(i=1;i<=n;i++)

	{

		l=0,r=N+1;

		while(l<r)

		{

			mid=(l+r)>>1;

			if(calc(i,mid)>=m)	r=mid;

			else	l=mid+1;

		}

		printf("%d\n",r);

	}

	return 0;

}