bzoj1316

题目链接:bzoj1316
题意:
现在给你一个n个点的树，给你q个询问，每次询问，询问树上是否有一条长度为k的路径。
n<=10000,q<=100.
题解:
我们发现q是极小的，那么我们只要每次询问能够做到的复杂度限制就不会太高，我们不妨考虑只有一个询问，要知道树上是否有一条长度为k的路径，我们不妨用点分治的方法来对路径进行分类，这样我们只要能够在较低的复杂度下处理点分治时的查询即可，我们可以点分治时，先将所有的dis(即这一层所有点到重心的距离)处理出来，放进一个set中，我们查询以某个节点为端点，经过重心的路径是否有长度为k的，这可以用set很容易做到，当然我们需要先把与这个节点在同一子层中的点给剔掉。这样此题就可以在nlogn^2的复杂度中得到解决。

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#include <cmath>
#include <algorithm>
#include <set> 
#define lld long long
using namespace std;
const int maxn=10010;
struct Edge{
    int to,next,w;
}edge[maxn*2];
int head[maxn],vis[maxn*2],p,n,use[maxn*2],line[maxn],size[maxn],op,fa[maxn],can[maxn],que[maxn],OP;
lld dis[maxn],l[maxn];
void build(int a,int b,int c){
    edge[++p]=(Edge){b,head[a],c};
    head[a]=p;
}
void dfs(int a){
    use[a]=1;
    size[a]=1;
    line[++op]=a;
    for (int i=head[a];i;i=edge[i].next)
        if (!vis[i]&&!use[edge[i].to]){
            fa[edge[i].to]=a;
            dfs(edge[i].to);
            size[a]+=size[edge[i].to];
        }
}
void dfs2(int a){
    use[a]=1;
    line[++op]=a;
    for (int i=head[a];i;i=edge[i].next)
        if (!vis[i]&&!use[edge[i].to]){
            dis[edge[i].to]=dis[a]+edge[i].w;
            fa[edge[i].to]=a;
            dfs2(edge[i].to);
        }   
}
void dfs3(int a){
    que[++OP]=a;
    for (int i=head[a];i;i=edge[i].next)
        if (!vis[i]&&fa[edge[i].to]==a)
            dfs3(edge[i].to);
}
int store[maxn];
int get(int a){
    op=0;
    fa[a]=0;
    dfs(a);
    int ans=a;
    for (int i=1;i<=op;i++){
        store[line[i]]=size[a]-size[line[i]];
        for (int j=head[line[i]];j;j=edge[j].next)
            if (!vis[j]&&edge[j].to!=fa[line[i]])
                store[line[i]]=max(store[line[i]],size[edge[j].to]);
        if (store[line[i]]<store[ans])
            ans=line[i];
        use[line[i]]=0;
    }
    return ans;
}
set<lld> s;
int q;
void fenzhi(int a){
    s.clear();
    int point=get(a);
    op=0;
    fa[point]=0;
    dis[point]=0;
    dfs2(point);
    for (int i=1;i<=op;i++)
        s.insert(dis[line[i]]);
    for (int i=head[point];i;i=edge[i].next)    
        if (!vis[i]){
            OP=0;
            dfs3(edge[i].to);
            for (int j=1;j<=OP;j++)
                s.erase(dis[que[j]]);
            set<lld>::iterator it;
            for (int j=1;j<=OP;j++)
                for (int k=1;k<=q;k++)
                    if (!can[k]){
                        it=s.find(l[k]-dis[que[j]]);
                        if (it!=s.end())
                            can[k]=1;
                    }
            for (int j=1;j<=OP;j++)     
                s.insert(dis[que[j]]);
        }
    for (int i=1;i<=op;i++)         
        use[line[i]]=0;
    for (int i=head[point];i;i=edge[i].next)
        if (!vis[i]){
            vis[i]=1;
            vis[i^1]=1;
            fenzhi(edge[i].to);
        }
}
int main(){
    freopen("1316.in","r",stdin);
    freopen("1316.out","w",stdout);
    scanf("%d%d",&n,&q);
    p=1;
    for (int i=1;i<=n-1;i++){
        int x,y,c;
        scanf("%d%d%d",&x,&y,&c);
        build(x,y,c);
        build(y,x,c);
    }
    for (int i=1;i<=q;i++) scanf("%lld",&l[i]);
    fenzhi(1);
    for (int i=1;i<=q;i++){
        if (!can[i]&&l[i]!=0)
            printf("No");
        else printf("Yes");
        printf("\n");
    }
    fclose(stdin);
    fclose(stdout);
}