洛谷 P3806 点分治1

题目链接

题意

判断树上距离为k的点对是否存在，树边有权值

思路

点分治模板题，将点对的路径按经过某点和不经过某点分治。
每次选重心分治，每次求解复杂度为未枚举点的连通块大小，如果是条链大概 $n+2\ast n/2+4\ast n/4\dots \dots$，用心感受下大概就是O(n)*递归层数，那么链应该是最坏复杂度，本题总的复杂度就是 $O(n\ast m\ast logn)$

点分治大概思路就是 getrt 找重心，dfs 枚举当前选中的重心，calc 计算经过该重心的解。

代码

#include <bits/stdc++.h>
using namespace std;

#define N 10005
#define maxn 10000005

int tot, first[N];
struct Edge {
    int v, w, nxt;
}e[N<<1];

void add(int u, int v, int w) {
    e[++tot].v = v;
    e[tot].w = w;
    e[tot].nxt = first[u];
    first[u] = tot;
}
// sum = n总点数, sz[i]以i为根节点的子树大小， maxsz[i] 表示以i为根节点最大子树大小
// vis[i] 表示分治是否枚举过i为重心， rt表示当前枚举重心
int sum, sz[N], maxsz[N], vis[N], rt;
// dis[i]求i到rt的距离，las和now上次和当前情况，q记录las加了哪些方便删除
int dis[N], las[maxn], now[N], q[N];

void getrt(int u, int fa) {
    sz[u] = 1, maxsz[u] = 0;
    for(int i = first[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        if(v == fa || vis[v]) continue;
        getrt(v,u);
        sz[u] += sz[v];
        maxsz[u] = max(maxsz[u], sz[v]);
    }
    maxsz[u] = max(maxsz[u], sum-sz[u]);
    if(maxsz[u] < maxsz[rt]) rt = u;
}

void getdis(int u, int fa) {
    now[++now[0]] = dis[u];
//    printf("%d ##\n",dis[u]);
    for(int i = first[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        int w = e[i].w;
        if(v == fa || vis[v]) continue;
        dis[v] = dis[u]+w;
        getdis(v,u);
    }
}

int calc(int u, int k) {
    int p = 0;
    for(int i = first[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        if(vis[v]) continue;
        now[0] = 0;
        dis[v] = e[i].w;
        getdis(v,u);
        for(int j = now[0]; j; --j) {
            if(k-now[j] >= 0 && las[k-now[j]]) {
                for(int ii = 1; ii <= p; ++ii) las[q[ii]] = 0;
//                printf("!!!!!!\n");
                return 1;
            }
        }
        for(int j = now[0]; j; --j) q[++p] = now[j], las[now[j]] = 1;
    }
    for(int i = 1; i <= p; ++i) las[q[i]] = 0;
    return 0;
}

int dfs(int u, int k) {
//    printf("%d -------\n",u);
    vis[u] = las[0] = 1;
    if(calc(u, k)) return 1;
    for(int i = first[u]; i; i = e[i].nxt) {
        int v = e[i].v;
        if(vis[v]) continue;
        maxsz[rt = 0] = sum = sz[v];
        getrt(v, 0);
        if(dfs(rt, k)) return 1;
    }
    return 0;
}

int main() {
    int n, m;
    scanf("%d%d",&n,&m);
    for(int i = 1; i < n; ++i) {
        int u, v, w;
        scanf("%d%d%d",&u,&v,&w);
        add(u, v, w);
        add(v, u, w);
    }
    for(int i = 1; i <= m; ++i) {
        int k;
        scanf("%d",&k);
        maxsz[rt = 0] = sum = n;
        memset(vis,0,sizeof(vis));
        getrt(1,0);
        printf("%s\n",dfs(rt, k) ? "AYE" : "NAY");
    }
    return 0;
}