BZOJ1460: Pku2114 Boatherds

题目链接：点这里

题目描述：给你一棵n个点的带权有根树，有p个询问，每次询问树中是否存在一条长度为Len的路径，如果是，输出Yes否输出No.

数据范围：\(n\le1e5\,,p\le100\,,长度\le1e5\)

Solution:

点分治裸题，没什么好讲的。不过注意当询问0时，答案应该是Yes。

Code:

#pragma GCC optimize(3)
#include<bits/stdc++.h>
#define ll long long
#define inf 2147483647
using namespace std;
const int N=1e4+1;
int rt,rtu,cnt,szt,head[N],sz[N],mx[N],rtt[N];
int n,m,k,ans,l,r,tot,dis[N],vis[N];
struct Edge{int nxt,to,val;}edge[N<<1];
void ins(int x,int y,int z){
    edge[++cnt].nxt=head[x];
    edge[cnt].to=y;edge[cnt].val=z;
    head[x]=cnt;
}
void getrt(int x,int f){
    sz[x]=1,mx[x]=0;
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y==f||vis[y]) continue;
        getrt(y,x);sz[x]+=sz[y];
        mx[x]=max(mx[x],sz[y]);
    }mx[x]=max(mx[x],szt-sz[x]);
    if(mx[x]<mx[rt]) rt=x;
}
void getdis(int x,int f,int d){
    dis[++tot]=d;
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(y==f||vis[y]) continue;
        getdis(y,x,d+edge[i].val);
    }
}
int find(int x,int l,int r){
	while(l<r){
		int mid=(l+r)>>1;
		if(dis[mid]+x<k) l=mid+1;
		else r=mid;
	}return l;
}
int calc(int x,int d){
    tot=0;getdis(x,0,d);
    sort(dis+1,dis+tot+1);
    int num=0,l=1,r=tot;
    while(l<r){
		if(dis[l]+dis[r]>k) --r;
		else if(dis[l]+dis[r]<k) ++l;
		else{
			if(dis[l]==dis[r]){
				num+=(r-l+1)*(r-l)/2;
				break;
			}
			int u1=l,u2=r;
			while(dis[u1]==dis[l]) ++u1;
			while(dis[u2]==dis[r]) --u2;
			num+=(u1-l)*(r-u2);l=u1,r=u2;
		}
	}return num;
}
void divide(int x){
    ans+=calc(x,0);vis[x]=1;
    for(int i=head[x];i;i=edge[i].nxt){
        int y=edge[i].to;
        if(!vis[y]){
            ans-=calc(y,edge[i].val);
            divide(rtt[++rtu]);
        }
    }
}
void getrtt(int x){
	vis[x]=1;
	for(int i=head[x];i;i=edge[i].nxt){
		int y=edge[i].to;
		if(!vis[y]){
			rt=0,szt=sz[y],getrt(y,0);
			rtt[++rtu]=rt;getrtt(rt);
		}
	}
}
int solve(){
	memset(vis,0,sizeof(vis));
    ans=0;divide(rtt[rtu=0]);
    return ans;
}
int read(){
	int x=0,f=1;char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')f=-f;ch=getchar();}
	while(isdigit(ch)){x=x*10+ch-48;ch=getchar();}
	return x*f;
}
int main(){
	n=read(),m=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read(),z=read();
        ins(x,y,z);ins(y,x,z);
    }
    szt=n;mx[rt]=inf;ans=0;
    getrt(1,0);rtt[0]=rt;
	getrtt(rt);
	for(int i=1;i<=m;i++){
		k=read();
		if(!k){puts("Yes");continue;}
		solve();
		if(ans) puts("Yes");
		else puts("No");
	}
    return 0;
}