BZOJ1316 树上的询问

点分治常数很大所以我们跑一遍即可。

用set记录。

一开始没有判q[i]==0惨惨了//

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int sum,n,rt,q[N],p,f[N],son[N],head[N],cnt;
bool v[N],ans[N];
struct node{
    int to,nex,w;
}e[N<<1];
void add(int x,int y,int w)
{
    e[++cnt].w=w;e[cnt].nex=head[x];e[cnt].to=y;head[x]=cnt;
}
set<int>s;
void getrt(int x,int fa)
{
    son[x]=1;f[x]=0;
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(v[y]||y==fa)continue;
        getrt(y,x);
        son[x]+=son[y];
        f[x]=max(f[x],son[y]);
    }
    f[x]=max(f[x],sum-f[x]);
    if(f[x]<f[rt])rt=x;
    return;
}
void calc(int x,int fa,int w)
{
    for(int i=1;i<=p;++i)
    {
        if(s.find(q[i]-w)!=s.end())ans[i]=1;
    }
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(v[y]||y==fa)continue;
        calc(y,x,w+e[i].w);
    }
}
void update(int x,int fa,int w)
{
    s.insert(w);
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(v[y]||y==fa)continue;
        update(y,x,w+e[i].w);
    }
}
void work(int x)
{
    v[x]=1;s.insert(0);
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(v[y])continue;
        calc(y,x,e[i].w);
        update(y,x,e[i].w);
    }
    s.clear();
    for(int i=head[x];i;i=e[i].nex)
    {
        int y=e[i].to;
        if(v[y])continue;
        rt=0;sum=son[y];
        getrt(y,x);
        work(rt);    
    }
}
int main()
{
    scanf("%d%d",&n,&p);
    int x,y,w;
    for(int i=1;i<n;++i)
    {
        scanf("%d%d%d",&x,&y,&w);
        add(x,y,w);add(y,x,w);
    }
    for(int i=1;i<=p;++i)
    {
        scanf("%d",&q[i]);
    }
    f[0]=sum=n;
    getrt(1,0);
    work(rt);
    for(int i=1;i<=p;++i)
    {
        if(ans[i]||!q[i])puts("Yes");
        else puts("No"); 
    }
}