P3806 【模板】点分治1

 题目 https://www.luogu.org/problemnew/show/P3806

题意 模板题

思路 模板题

#include <bits/stdc++.h>

using namespace std;
const int inf = 10000000;
const int maxn = 100010;
int n,m;
struct node
{
    int v,dis,nxt;
}E[maxn<<1];
int tot,head[maxn];
int maxp[maxn],size[maxn];//当前节点最大子树节点数
int dis[maxn],rem[maxn],judge[maxn]; //根节点当当前节点长度 当前根节点到某子树所以子节点长度 当前根节点之前所以子树出现的长度
int vis[maxn];//当前节点是否已被分治
int test[maxn],q[maxn];//记录答案 记录出现的长度
int query[1100];
int sum,rt,ans;

void add(int u,int v,int dis)
{
    E[++tot].nxt = head[u];
    E[tot].v = v;
    E[tot].dis = dis;
    head[u] = tot;
}

void getrt(int u,int pa) //查询重心
{
    size[u] = 1;
    maxp[u] = 0;
    for(int i = head[u];i != -1;i = E[i].nxt)
    {
        int v = E[i].v;
        if(v == pa||vis[v]) continue;
        getrt(v,u);
        size[u] += size[v];
        maxp[u] = max(maxp[u],size[v]);
    }
    maxp[u] = max(maxp[u],sum-size[u]);
    if(maxp[u] < maxp[rt]) rt = u;
}
void getdis(int u,int fa) //查询节点到当前根的长度
{
    rem[++rem[0]] = dis[u];
    for(int i = head[u];i != -1;i=E[i].nxt)
    {
        int v = E[i].v;
        if(v == fa||vis[v]) continue;
        dis[v] = dis[u] + E[i].dis;
        getdis(v,u);
    }
}

void calc(int u) //当前根节点出现的结果
{
    int p = 0;
    for(int i = head[u];i != -1;i=E[i].nxt)
    {
        int v = E[i].v;
        if(vis[v]) continue;
        rem[0] = 0;
        dis[v] = E[i].dis;
        getdis(v,u);

        for(int j = rem[0];j >= 1;j--)
        {
            for(int k = 1;k <= m;k++)
            {
                if(query[k] >= rem[j])
                {
                    test[k]|=judge[query[k]-rem[j]];
                }
            }
        }
        for(int j = rem[0];j >= 1;j--)
        {
            q[++p] = rem[j],judge[rem[j]] = 1;
        }
    }
    for(int i = 1;i <= p;i++)
    {
        judge[q[i]] = 0;
    }
}

void solve(int u)
{
    vis[u] = judge[0] = 1;
    calc(u);
    for(int i = head[u];i != -1;i = E[i].nxt)
    {
        int v = E[i].v;
        if(vis[v]) continue;
        sum = size[v],maxp[rt = 0] = inf;
        getrt(v,0),solve(rt);
    }
}
int  main()
{
    memset(head,-1,sizeof(head));
    tot = 0;
    scanf("%d%d",&n,&m);
    for(int i = 1;i < n;i++) // 建图
    {
        int u,v,dis;
        scanf("%d%d%d",&u,&v,&dis);
        add(u,v,dis),add(v,u,dis);
    }
    for(int i = 1;i <= m;i++)
    {
        scanf("%d",&query[i]);
    }
    maxp[rt] = sum = n;
    getrt(1,0); //求重心
    memset(vis,0,sizeof(vis));
    solve(rt);//开始分治
    for(int i =1;i <= m;i++)
    {
        if(test[i]) printf("AYE\n");
        else printf("NAY\n");
    }
    return 0;
}