bzoj1316 树上的询问 点分治

原题:https://www.lydsy.com/JudgeOnline/problem.php?id=1316

题解:这题几乎和洛谷的题一样，但bzoj上不能把数组开到1000000。。。我开了结果re了所以我们在处理有没有相同的时候用二分的思路。排完序枚举长度，lower/upper查找区间就行。

#include<bits/stdc++.h>
#define inf (1<<31)-1 
using namespace std;
const int N=10010;
struct E{int to,nxt,c;}data[N<<1];
int len,n,m,rt;
int d[N],tmp[N],f[N],sz[N],h[N],q[101],a[101];
bool vis[N];
inline int rd(){
    int x=0;int f=1;char s=getchar();
    while(!isdigit(s)) f=(s=='-'?-1:f),s=getchar();
    while(isdigit(s))  x=(x<<1)+(x<<3)+s-'0',s=getchar();
    return x*f;
}

inline void ins(int x,int y,int c){
    data[++len].to=y;data[len].c=c;data[len].nxt=h[x];h[x]=len;
    data[++len].to=x;data[len].c=c;data[len].nxt=h[y];h[y]=len;
}

void getroot(int x,int fa){
    f[x]=0;sz[x]=1;
    for(int i=h[x];i;i=data[i].nxt){
        int y=data[i].to;if(vis[y] || y==fa ) continue;
        getroot(y,x);
        sz[x]+=sz[y];
        f[x]=max(f[x],sz[y]);
    }
    f[x]=max(f[x],n-sz[x]);
    if(f[rt]>f[x]) rt=x;
}
void dfs(int x,int fa){
	tmp[++tmp[0]]=d[x];
    for(int i=h[x];i;i=data[i].nxt){
        int y=data[i].to;if(vis[y] || y==fa) continue;
        d[y]=d[x]+data[i].c;
        dfs(y,x);
    }
}
void work(int x,int val,int c){
    d[x]=val;tmp[0]=0;
    dfs(x,0); 
    sort(tmp+1,tmp+1+tmp[0]);
    for(int k=1;k<=m;k++){
        for(int i=1;i<=tmp[0];i++){
            if(2*tmp[i]>q[k]) break;
            int r=upper_bound(tmp+1,tmp+tmp[0]+1,q[k]-tmp[i])-(tmp+1);
			int l=lower_bound(tmp+1,tmp+tmp[0]+1,q[k]-tmp[i])-(tmp+1); 
            a[k]+=c*(r-l);
        }
    }
}
void sol(int x){
    work(x,0,1);
    vis[x]=1;
    for(int i=h[x];i;i=data[i].nxt){
        int y=data[i].to;if(vis[y])continue;
        work(y,data[i].c,-1);
        n=sz[y];rt=0;getroot(y,0);
        sol(rt);
    }
}



int main(){
    
//	freopen("testdata.in","r",stdin);
    len=1;memset(h,0,sizeof h);memset(vis,0,sizeof vis); 
    rt=0;f[0]=inf;
    
    
    n=rd();m=rd();
    for(int i=1,x,y,c;i<n;i++){
        x=rd();y=rd();c=rd();ins(x,y,c);
    }
    for(int i=1;i<=m;i++) q[i]=rd(); 
    
    getroot(1,0);
    sol(rt);
	for(int i=1;i<=m;i++) printf("%s\n",(a[i]|| !q[i])?"Yes":"No");
 //   for(int i=1;i<=m;i++) printf("%s\n",a[i]?"AYE":"NAY");
    return 0;
}