BZOJ 2117: [2010国家集训队]Crash的旅游计划 动态点分治+二分

感觉现在写点分治可快了~

二分答案，就可以将求第 $k$ 大转换成一个判断问题，直接拿点分树判断一下就行了. 

#include <cstdio> 
#include <vector> 
#include <algorithm>  
#define N 100004 
#define setIO(s) freopen(s".in","r",stdin) 
using namespace std;   
int n,edges,K; 
int hd[N],to[N<<1],nex[N<<1],val[N<<1];  
void add(int u,int v,int c)
{
	nex[++edges]=hd[u],hd[u]=edges,to[edges]=v,val[edges]=c; 
}   
namespace tree
{ 
	int size[N],son[N],fa[N],top[N],dep[N],dis[N]; 
	void dfs1(int u,int ff)
	{ 
		fa[u]=ff,size[u]=1; 
		for(int i=hd[u];i;i=nex[i]) 
			if(to[i]!=ff) 
			{ 
				dep[to[i]]=dep[u]+1,dis[to[i]]=dis[u]+val[i]; 
				dfs1(to[i],u); 
				size[u]+=size[to[i]]; 
				if(size[to[i]]>size[son[u]]) son[u]=to[i]; 
			}
	} 
	void dfs2(int u,int tp)
	{    
		top[u]=tp; 
		if(son[u]) dfs2(son[u],tp); 
		for(int i=hd[u];i;i=nex[i]) 
			if(to[i]!=fa[u]&&to[i]!=son[u]) 
				dfs2(to[i],to[i]);      
	}
	int LCA(int x,int y)
	{
		while(top[x]!=top[y]) 
			dep[top[x]]>dep[top[y]]?x=fa[top[x]]:y=fa[top[y]]; 
		return dep[x]<dep[y]?x:y;    
 	} 
 	int Dis(int x,int y)
 	{
 		return dis[x]+dis[y]-(dis[LCA(x,y)]<<1);    
 	} 
};  
vector<int>F[N],G[N];     
int root,sn; 
int mx[N],size[N],vis[N],Fa[N];      
void dfs(int u,int ff)
{
	size[u]=1; 
	for(int i=hd[u];i;i=nex[i]) 
		if(!vis[to[i]]&&to[i]!=ff) 
			dfs(to[i],u),size[u]+=size[to[i]]; 
} 
void getroot(int u,int ff)
{
	size[u]=1,mx[u]=0; 
	for(int i=hd[u];i;i=nex[i]) 
		if(to[i]!=ff&&!vis[to[i]])  
			getroot(to[i],u),size[u]+=size[to[i]],mx[u]=max(mx[u],size[to[i]]); 
	mx[u]=max(mx[u],sn-size[u]);  
	if(mx[u]<mx[root]) root=u; 
}  
void calc(int u,int ff,int dep,int rt) 
{  
	F[rt].push_back(dep);     
	if(Fa[rt]) G[rt].push_back(tree::Dis(Fa[rt], u));          
	for(int i=hd[u];i;i=nex[i]) 
		if(to[i]!=ff&&!vis[to[i]]) 
			calc(to[i],u,dep+val[i],rt);     
}
void prepare(int u)
{
	vis[u]=1; 
	calc(u,0,0,u);    
	F[u].push_back(1000000004);       
	sort(F[u].begin(),F[u].end()); 
	if(Fa[u]) 
	{
		G[u].push_back(1000000004); 
		sort(G[u].begin(),G[u].end());     
	}
	for(int i=hd[u];i;i=nex[i]) 
		if(!vis[to[i]]) 
			dfs(to[i],u),sn=size[to[i]],root=0,getroot(to[i],u),Fa[root]=u,prepare(root);     
} 
int query(int u,int k) 
{   
	int U=u, re=upper_bound(F[u].begin(),F[u].end(),k)-F[u].begin()-1;       
	for(;Fa[u];u=Fa[u]) 
	{
		int dis=tree::Dis(Fa[u],U); 
		if(dis<=k) 
		{
			re+=(upper_bound(F[Fa[u]].begin(),F[Fa[u]].end(),k-dis)-F[Fa[u]].begin());
			re-=(upper_bound(G[u].begin(),G[u].end(),k-dis)-G[u].begin()); 
		}
	}           
	return re;    
}
int main() 
{
	int i,j,tot=0; 
	char ss[2]; 
	// setIO("input"); 
	scanf("%s%d%d",ss,&n,&K);  
	for(i=1;i<n;++i) 
	{
		int a,b,c; 
		scanf("%d%d%d",&a,&b,&c),add(a,b,c),add(b,a,c),tot+=c; 
 	} 
 	tree::dfs1(1,0);  
 	tree::dfs2(1,1); 
 	mx[0]=sn=n,root=0,getroot(1,0),prepare(root);
  	for(i=1;i<=n;++i) 
 	{
 		int l=1,r=tot,mid,ans=0; 
 		while(l<=r) 
 		{
 			mid=(l+r)>>1; 
 			if(query(i,mid)>=K) 
 			{
 				ans=mid,r=mid-1; 
 			}
 			else l=mid+1;       
 		}
 		printf("%d\n",ans);     
 	}
	return 0; 
}