1316: 树上的询问（点分治）

1316: 树上的询问

链接

分析

　　每次查找出重心（去掉重心后的最大的联通块最小，保证复杂度），然后统计过重心的路径中有没有长度等于len的。

　　统计时，由于必须要过重心，不能是同一棵子树中的。可以挨个遍历每棵子树，然后统计即可。

　　判断时，用set查找一下即可。

代码

#include<cstdio>
#include<algorithm>
#include<cctype>
#include<set>
 
using namespace std;
 
const int N = 10010;
 
struct Edge{
    int to,nxt,w;
    Edge() {}
    Edge(int a,int b,int c) {to = a,w = b,nxt = c;}
}e[N<<1];
int A[N],q[N],ans[N],dis[N],deth[N],head[N],siz[N];
int tot,Root,Size,Cnt,Num = 1e9,n,m;
bool vis[N];
set<int>s;
 
inline int read() {
    int x = 0,f = 1;char ch=getchar();
    for (; !isdigit(ch); ch=getchar()) if(ch=='-')f=-1;
    for (; isdigit(ch); ch=getchar()) x=x*10+ch-'0';
    return x*f;
}
void add_edge(int u,int v,int w) {
    e[++tot] = Edge(v,w,head[u]);head[u] = tot;
    e[++tot] = Edge(u,w,head[v]);head[v] = tot;
}
void getRoot(int u,int fa) {
    siz[u] = 1;
    int mx = 0;
    for (int i=head[u]; i; i=e[i].nxt) {
        int v = e[i].to;
        if (v == fa || vis[v]) continue;
        getRoot(v,u);
        siz[u] += siz[v];
        mx = max(siz[v],mx);
    }
    mx = max(mx,Size-siz[u]);
    if (mx < Num) Root = u,Num = mx;
}   
void dfs(int u,int fa) {
    A[++Cnt] = dis[u];
    for (int i=head[u]; i; i=e[i].nxt) {
        int v = e[i].to;
        if (v == fa || vis[v]) continue;
        dis[v] = dis[u] + e[i].w;
        dfs(v,u);
    }
}
void calcc(int u) {
    s.clear();
    s.insert(0);
    dis[u] = 0;
    for (int i=head[u]; i; i=e[i].nxt) {
        int v = e[i].to;
        if (vis[v]) continue;
        Cnt = 0;
        dis[v] = e[i].w;
        dfs(v,u);
        for (int j=1; j<=Cnt; ++j) {
            for (int k=1; k<=m; ++k) {
                if (ans[k]) continue;
                if (s.find(q[k]-A[j])!=s.end()) ans[k] = 1;
            }
        }
        for (int j=1; j<=Cnt; ++j) s.insert(A[j]);
    }
}
void solve(int u) {
    vis[u] =  true;
    calcc(u);
    for (int i=head[u]; i; i=e[i].nxt) {
        int v = e[i].to;
        if (vis[v]) continue;
        Size = siz[v];Num = 1e9;Root = 0;
        getRoot(v,0);
        solve(Root);
    }
}
int main() {
    n = read(),m = read();
    for (int u,v,w,i=1; i<n; ++i) {
        u = read(),v = read(),w = read();
        add_edge(u,v,w);
    }
    for (int i=1; i<=m; ++i) {
        q[i] = read();
        if (!q[i]) ans[i] = 1;
    }
    Size = n;
    Num = 1e9;
    getRoot(1,0);
    solve(Root);
    for (int i=1; i<=m; ++i) 
        if (ans[i]) puts("Yes");
        else puts("No");
    return 0;
}