bzoj 1316（点分治）

传送门
题解：点分治后用set/map查找，查找方式以前还真没用过：每句一个dis，用一个被询问的距离q[i]去减dis，看减出来的值是否出现过。
好像也可以用二分（zyf2000）但是比较慢。。。

#include<bits/stdc++.h>using namespace std;const int MAXN=1e4+5,INF=0x3f3f3f3f;int n,m,q[102];int head[MAXN],edge=0;struct EDGE {    int v,nxt,w;}e[MAXN<<1];int vis[MAXN],ans[102];int siz[MAXN],mx[MAXN],dis[MAXN],a[MAXN],root,sum,tot;set<int> s;inline int read() {    int x=0,f=1;char c=getchar();    while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();}    while (c>='0'&&c<='9') x=x*10+c-'0',c=getchar();    return x*f;}inline void adde(int u,int v,int w) {    e[edge].nxt=head[u],e[edge].v=v,e[edge].w=w,head[u]=edge++;    e[edge].nxt=head[v],e[edge].v=u,e[edge].w=w,head[v]=edge++;}void getroot(int p,int fa) {    siz[p]=1,mx[p]=0;    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (vis[v]||v==fa) continue;        getroot(v,p);        siz[p]+=siz[v];        mx[p]=max(mx[p],siz[v]);    }    mx[p]=max(mx[p],sum-siz[p]);    if (mx[p]<mx[root]) root=p;}void getdis(int p,int fa) {    siz[p]=1,a[++tot]=dis[p];    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (vis[v]||v==fa) continue;        dis[v]=dis[p]+e[i].w;        getdis(v,p);        siz[p]+=siz[v];    }}void work(int p) {    vis[p]=true,s.clear(),s.insert(0),dis[p]=0;    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (!vis[v]) {            tot=0,dis[v]=e[i].w,getdis(v,p);            for (int j=1;j<=tot;++j)                for (int k=1;k<=m;++k)                    if (s.find(q[k]-a[j])!=s.end()) ans[k]=1;            for (int j=1;j<=tot;++j) s.insert(a[j]);        }    }    for (int i=head[p];~i;i=e[i].nxt) {        int v=e[i].v;        if (!vis[v]) {            sum=siz[v];            root=0;            getroot(v,0);            work(root);        }    }}int main() {//  freopen("bzoj 1316.in","r",stdin);    memset(vis,false,sizeof(vis));    memset(ans,false,sizeof(ans));    memset(head,-1,sizeof(head));    n=read(),m=read();    for (int i=1;i<n;++i) {        int u=read(),v=read(),w=read();        adde(u,v,w);    }    for (int i=1;i<=m;++i) q[i]=read();    sum=n,root=0;    mx[0]=INF;    getroot(1,0);    work(root);    for (int i=1;i<=m;++i)        puts((ans[i]||q[i]==0)?"Yes":"No");    return 0;}