习题：点分治（淀粉质）

Posted 2021-03-23 loney-s

题目

传送门

思路

如题目所示，就是一道淀粉质的版题

代码

#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
using namespace std;
#define MAXN 100000
#define MAXM 200000
int n,m;
int cnt;
int focus;
int s;
int head[MAXN+5];
int val[MAXN+5];
int nxt[MAXM+5];
int to[MAXM+5];
int siz[MAXN+5];
int maxx[MAXN+5];
int dis[MAXN+5];
int ans[10000005];
int v[MAXN];
int lenv;
bool vis[MAXN+5];
void add_edge(int u,int v,int w)
{
    to[++cnt]=v;
    val[cnt]=w;
    nxt[cnt]=head[u];
    head[u]=cnt;
}
void find_focus(int u,int fa)
{
    siz[u]=1;
    maxx[u]=0;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(!vis[v]&&v!=fa)
        {
            find_focus(v,u);
            siz[u]+=siz[v];
            maxx[u]=max(maxx[u],siz[v]);
        }
    }
    maxx[u]=max(maxx[u],s-siz[u]);
    if(maxx[u]<maxx[focus])
        focus=u;
}
void solve_dis(int u,int fa)
{
    v[lenv++]=dis[u];
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(!vis[v]&&v!=fa)
        {
            dis[v]=dis[u]+val[i];
            solve_dis(v,u);
        }
    }
}
void solve(int s,int l,int w)
{
    lenv=0;
    dis[s]=l;
    solve_dis(s,0);
    for(int i=0;i<lenv;i++)
        for(int j=i+1;j<lenv;j++)
            if(v[i]+v[j]<=10000000)
                ans[v[i]+v[j]]+=w;

}
void dfs(int u)
{
    solve(u,0,1);
    vis[u]=1;
    for(int i=head[u];i;i=nxt[i])
    {
        int v=to[i];
        if(!vis[v])
        {
            solve(v,val[i],-1);
            focus=0;
            //maxx[0]=n+1;
            s=siz[u];
            find_focus(v,u);
            dfs(focus);
        }
    }
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>n>>m;
    for(int i=1,u,v,w;i<n;i++)
    {
        cin>>u>>v>>w;
        add_edge(u,v,w);
        add_edge(v,u,w);
    }
    s=n;
    maxx[0]=n+1;
    find_focus(1,0);
    dfs(focus);
    for(int i=1,k;i<=m;i++)
    {
        cin>>k;
        if(ans[k])
            cout<<"AYE
";
        else
            cout<<"NAY
";
    }
    return 0;
}