码迷,mamicode.com
首页 > 其他好文 > 详细

P3806 【模板】点分治1

时间:2019-06-02 17:56:20      阅读:110      评论:0      收藏:0      [点我收藏+]

标签:++   base   play   说明   hit   mes   names   dfs   ttl   

题目背景

感谢hzwer的点分治互测。

题目描述

给定一棵有n个点的树

询问树上距离为k的点对是否存在。

输入输出格式

输入格式:

n,m 接下来n-1条边a,b,c描述a到b有一条长度为c的路径

接下来m行每行询问一个K

输出格式:

对于每个K每行输出一个答案,存在输出“AYE”,否则输出”NAY”(不包含引号)

输入输出样例

输入样例#1: 
2 1
1 2 2
2
输出样例#1: 
AYE

说明

对于30%的数据n<=100

对于60%的数据n<=1000,m<=50

对于100%的数据n<=10000,m<=100,c<=10000,K<=10000000

代码

对于每个点,分为有长度为k的路径经过点以及长度为k的路径在其子树

对于长度为k的路径在其子树的点,我们可以将其看作是长度为k的路径经过其子树所在根节点

那么这就是点分治的基本思想

为降低复杂度,我们每次采取重心作为根节点。将树的深度将至logn

则复杂度为O(NlogN2)

 

技术图片
#include<bits/stdc++.h>
#define inf 0x3f3f3f3f
using namespace std;
const int maxn=10000+100,maxm=10000000+100;
int head[maxn];
int q[maxn],ans[maxn];
int p[maxm],t[maxn];
int d[maxn],dis[maxn],vis[maxn];
int s[maxn],ms[maxn];
int rt=0,sum;
int n,m;
struct edge
{
    int to,next,val;
}e[maxn<<1];
int size=0;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<0||ch>9){if(ch==-)f=-1;ch=getchar();}
    while(ch>=0&&ch<=9){x=(x<<3)+(x<<1)+ch-0;ch=getchar();}
    return x*f;
}
void addedge(int u,int v,int w)
{
    e[++size].to=v;e[size].val=w;e[size].next=head[u];head[u]=size;
}
void tc(int u,int fa)//树的重心 
{
    s[u]=1;ms[u]=0;
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to==fa||vis[to])continue;
        tc(to,u);
        s[u]+=s[to];
        ms[u]=max(ms[u],s[to]);
    }
    ms[u]=max(ms[u],sum-ms[u]);
    if(ms[u]<ms[rt])rt=u;
}
void dfs(int u,int fa)
{
    d[++d[0]]=dis[u];
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].to;
        if(to==fa||vis[to])continue;
        dis[to]=dis[u]+e[i].val;
        dfs(to,u);
    }
}
void cal(int u)
{
    int tot=0;
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].to;
        if(vis[to])continue;
        d[0]=0;dis[to]=e[i].val;
        dfs(to,u);
        for(int j=1;j<=d[0];++j)
        for(int k=1;k<=m;++k)
        if(!ans[k]&&q[k]>=d[j])//有长度为q[k]的路径:该子树与之前其他子树距离和为q[k]
        ans[k]=p[q[k]-d[j]];
        for(int j=1;j<=d[0];++j)
        t[++tot]=d[j],p[d[j]]=1;//用桶存下到该节点已有的距离 
    }
    for(int i=1;i<=tot;i++)
    p[t[i]]=0;//桶清空,用memsetTLE 
}
void solve(int u)
{
    vis[u]=p[0]=1;cal(u);
    for(int i=head[u];i;i=e[i].next)
    {
        int to=e[i].to;
        if(vis[to])continue;
        sum=s[to],ms[rt=0]=inf;
        tc(to,0);solve(rt);
    }
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read(),w=read();
        addedge(u,v,w);
        addedge(v,u,w);
    }
    for(int i=1;i<=m;i++)
    q[i]=read();
    sum=n;ms[rt]=inf;
    tc(1,0);
    solve(rt);
    for(int i=1;i<=m;i++)
    puts(ans[i]?"AYE":"NAY");
    return 0;
}
View Code

 

 

 

P3806 【模板】点分治1

标签:++   base   play   说明   hit   mes   names   dfs   ttl   

原文地址:https://www.cnblogs.com/DriverBen/p/10963438.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!