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;
}