bzoj 1316: 树上的询问 （点分治）

1316: 树上的询问

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 564   Solved: 150
[ Submit][ Status][ Discuss]

Description

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

Input

第一行两个整数n, p分别表示点的个数和询问的个数． 接下来n-1行每行三个数x, y, c，表示有一条树边x→y，长度为c． 接下来p行每行一个数Len，表示询问树中是否存在一条长度为Len的路径．

Output

输出有p行，Yes或No.

Sample Input

6 4
1 2 5
1 3 7
1 4 1
3 5 2
3 6 3
1
8
13
14

Sample Output

Yes
Yes
No
Yes

HINT

30%的数据，n≤100． 
100%的数据，n≤10000，p≤100，长度≤1000000． 

做完此题可看下POJ 3237 Tree

Source

[ Submit][ Status][ Discuss]

题解：点分治

#include<iostream>  
#include<cstdio>  
#include<algorithm>  
#include<cmath>  
#include<cstring>  
#define N 30003  
using namespace std;  
int n,m,tot,num[10000003];  
int point[N],next[N],v[N],len[N],cnt,root,sum,vis[N];  
int f[N],deep[N],mp[N],d[N],ak[N],mark[N],son[N],g[N];  
void add(int x,int y,int z)  
{  
    tot++; next[tot]=point[x]; point[x]=tot; v[tot]=y; len[tot]=z;  
    tot++; next[tot]=point[y]; point[y]=tot; v[tot]=x; len[tot]=z;  
    //cout<<x<<" "<<y<<" "<<z<<endl;  
}  
void getroot(int x,int fa)  
{  
    son[x]=1; f[x]=0;  
    for (int i=point[x];i;i=next[i]){  
        if (vis[v[i]]||v[i]==fa) continue;  
        getroot(v[i],x);  
        son[x]+=son[v[i]];  
        f[x]=max(f[x],son[v[i]]);  
    }  
    f[x]=max(f[x],sum-son[x]);  
    if (f[x]<f[root]) root=x;  
}  
void getdeep(int x,int fa)  
{  
    deep[++deep[0]]=d[x];  
    for (int i=point[x];i;i=next[i]){  
        if (vis[v[i]]||v[i]==fa) continue;  
        d[v[i]]=d[x]+len[i];  
        getdeep(v[i],x);  
    }  
}  
void cal(int x,int now,int opt)  
{  
    d[x]=now; deep[0]=0;  
    getdeep(x,0);  
    sort(deep+1,deep+deep[0]+1);  
    g[0]=0;  
    for (int i=1;i<=deep[0];i++)  
     if (deep[i]!=deep[i-1]||i==1)   
      g[++g[0]]=deep[i],num[g[0]]=1;  
     else num[g[0]]++;  
    for (int i=1;i<=m;i++) {  
       int l=1; int r=g[0];  
       for (int j=1;j<=g[0];j++)   
        if (num[j]>1&&g[j]*2==ak[i])  
         mark[i]+=opt*num[j]*(num[j]-1);   
       while (l<r) {  
          if (g[l]+g[r]<ak[i]) l++;  
          else {  
             if (g[l]+g[r]==ak[i]) mark[i]+=opt*num[l]*num[r];  
             r--;  
           }  
       }  
    }  
}  
void work(int x)  
{  
    cal(x,0,1);  
    vis[x]=1;  
    for (int i=point[x];i;i=next[i]){  
        if (vis[v[i]]) continue;  
        cal(v[i],len[i],-1);  
        sum=son[v[i]]; root=0;  
        getroot(v[i],x);  
        work(root);  
    }  
}  
int main()  
{  
    freopen("a.in","r",stdin);  
    freopen("my.out","w",stdout);  
    scanf("%d%d",&n,&m);  
    for (int i=1;i<n;i++) {  
        int x,y,z; scanf("%d%d%d",&x,&y,&z);  
        add(x,y,z);  
    }  
    for (int i=1;i<=m;i++) scanf("%d",&ak[i]);  
    sum=n; f[0]=1000000000; root=0;  
    getroot(1,0);  
    work(root);  
    for (int i=1;i<=m;i++)  
     if(mark[i]>0||!ak[i]) printf("Yes\n");  
     else printf("No\n");  
}