[bzoj1316] 树上的询问

　　裸的点分治。。

　　及时把已经确定的询问清掉就能快不少。时间复杂度O(nlogn*p)

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10023;
struct zs{int len,id;bool done;}q[103];
struct zs1{int too,pre,dis;}e[maxn<<1];int tot,last[maxn];
int st[maxn],top,pre,sz[maxn],mx[maxn],POI,dis[maxn];
bool gg[103],can[1000023],del[maxn];
int i,j,k,n,m,rt,mm;
 
int ra;char rx;
inline int read(){
    rx=getchar(),ra=0;
    while(rx<'0'||rx>'9')rx=getchar();
    while(rx>='0'&&rx<='9')ra*=10,ra+=rx-48,rx=getchar();return ra;
}
inline int max(int a,int b){return a>b?a:b;}
void getrt(int x,int fa){
    sz[x]=1,mx[x]=0;
    for(int i=last[x];i;i=e[i].pre)if(!del[e[i].too]&&e[i].too!=fa)
        getrt(e[i].too,x),sz[x]+=sz[e[i].too],mx[x]=max(mx[x],sz[e[i].too]);
    mx[x]=max(mx[x],POI-sz[x]);
    if(mx[x]<mx[rt])rt=x;
}
void getpoi(int x,int fa){
    st[++top]=dis[x];
    for(int i=last[x];i;i=e[i].pre)if(!del[e[i].too]&&e[i].too!=fa)
        dis[e[i].too]=dis[x]+e[i].dis,getpoi(e[i].too,x);
}
inline void updQ(){
    mm=m;m=0;
    for(int i=1;i<=mm;i++)if(!q[i].done)q[++m]=q[i];
}
void work(int x){
    int i,j,k;bool flag=0;
    rt=0,getrt(x,0);
    can[0]=1;pre=1;
    for(i=last[rt];i;i=e[i].pre)if(!del[e[i].too]){
        dis[e[i].too]=e[i].dis,getpoi(e[i].too,rt);
        for(j=pre;j<=top;j++)
            for(k=1;k<=m&&q[k].len>=st[j];k++)if(can[q[k].len-st[j]])q[k].done=1,flag=1;
        while(pre<=top)
            if(st[pre]<=q[1].len)
                can[st[pre++]]=1;
            else pre++;
    }
    if(flag)updQ();
    if(!m)return;
    while(top)
        if(st[top]<=1000000)can[st[top--]]=0;else top--;
    del[rt]=1;
    for(i=last[rt];i;i=e[i].pre)if(!del[e[i].too])POI=sz[e[i].too],work(e[i].too);
}
 
inline void insert(int a,int b,int c){
    e[++tot].too=b,e[tot].dis=c,e[tot].pre=last[a],last[a]=tot;
    e[++tot].too=a,e[tot].dis=c,e[tot].pre=last[b],last[b]=tot;
}
bool cmp(zs a,zs b){return a.len>b.len;}
 
int main(){mx[0]=1e9;
    n=read(),m=read();int m1=m;
    for(i=1;i<n;i++)j=read(),k=read(),insert(j,k,read());
    for(i=1;i<=m;i++)q[i].len=read(),q[i].id=i;
    sort(q+1,q+1+m,cmp);
    while(m&&q[m].len==0)q[m].done=1,m--;
    POI=n,work(1);
    for(i=1;i<=m;i++)gg[q[i].id]=1;
    for(i=1;i<=m1;i++)puts(gg[i]?"No":"Yes");
    return 0;
}