P3806 【模板】点分治1

题意：
给定一棵有 n 个点的树，询问树上距离为 k 的点对是否存在。

题解：
因为题目是要求路径长为k的路径条数，所以solve函数返回的是过x节点的长度为k的路径。
而这路径长度是可以用 O(n)O(n)O(n) 的方法求出
点分治，没啥好说的。

#include<bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
const int maxn=2e5+10;struct E{int to,w,next;
}edge[maxn];
int head[maxn],cnt;
int maxp[maxn],dis[maxn],sz[maxn];
bool visited[maxn],judge[10000000];
int sum,rt,q[maxn],nowcnt,temp[maxn];
int n,m;
void getrt(int x,int fa){  sz[x]=1,maxp[x]=0;//maxp初始化为最小值//遍历所有儿子，用maxp保留最大大小的儿子大小for(int i=head[x];~i;i=edge[i].next){int to=edge[i].to;//int w=edge[i].w;if(to==fa||visited[to]) continue;  //被删掉的也不算getrt(to,x);sz[x]+=sz[to];if(sz[to]>maxp[x]) maxp[x]=sz[to]; //更新maxp}maxp[x]=max(maxp[x],sum-sz[x]);if(maxp[x]<maxp[rt]) rt=x;
}void add(int u,int v,int w){edge[cnt].to=v;edge[cnt].w=w;edge[cnt].next=head[u];head[u]=cnt++;
}void getdis(int x,int fa){   temp[nowcnt++]=dis[x];for(int i=head[x];~i;i=edge[i].next){int v=edge[i].to;if(v==fa||visited[v]) continue;dis[v]=dis[x]+edge[i].w;getdis(v,x);}
}
int ans[105];
void sol(int x){queue<int> que;for(int i=head[x];~i;i=edge[i].next){int v=edge[i].to;if(visited[v]) continue;nowcnt=0;   //注意置零计数器dis[v]=edge[i].w;getdis(v,x);  //把距离都处理出来for(int j=0;j<nowcnt;j++){  //遍历所有距离for(int k=1;k<=m;k++){  //遍历所有询问if(q[k]>=temp[j]){  //如果询问大于单条路径的长度，就有可能存在ans[k]|=judge[q[k]-temp[j]];}}}for(int j=0;j<nowcnt;j++){  //把存在的单挑路径长度标为true，供下文使用que.push(temp[j]);judge[temp[j]]=true;}}while(!que.empty()){judge[que.front()]=false;que.pop();}
}
void divide(int x){  visited[x]=judge[0]=true;  //删除根sol(x);  //计算经过根节点的路径for(int i=head[x];~i;i=edge[i].next){int v=edge[i].to;if(visited[v]) continue;maxp[rt=0]=sum=sz[v];  //重心设为0，把maxp[0]至为最大值getrt(v,0);getrt(rt,0);  //与主函数相同divide(rt);}
}
signed main(){ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);memset(head,-1,sizeof head);memset(visited,false,sizeof visited);cin>>n>>m;sum=n;for(int i=1;i<n;i++){int u,v,w;cin>>u>>v>>w;add(u,v,w);add(v,u,w);}maxp[0]=sum=n;  //maxp[0]设为最大值getrt(1,0);  //找重心getrt(rt,0);  //此时siz数组存放的是1为根的时的大小，需要以找出的重心为根重算。for(int i=1;i<=m;i++) cin>>q[i];divide(rt);   //找好重心就可以分治了for(int i=1;i<=m;i++){if(ans[i]) cout<<"AYE"<<endl;else cout<<"NAY"<<endl;}
}

P3806 【模板】点分治1相关推荐

洛谷 - P4721 【模板】分治 FFT(分治NTT)
题目链接:点击查看题目大意:给出序列 g1,⋯,ng_{1,\cdots,n}g1,⋯,n,求 f0,⋯,nf_{0,\cdots,n}f0,⋯,n 规定 fi=∑j=1ifi−jgjf_i=\ ...
P4721 【模板】分治 FFT
传送门文章目录题意: 思路: 题意: 思路: 写一下式子,发现每个fif_ifi只有左边的fff对他有影响,所以考虑分治FFTFFTFFT来解决这个问题. 先递归左边,让后计算对右边贡献,再递归 ...
P3806来波点分治
分治: 一,一棵树上的路径可以分两种:1.经过根节点的:2.不经过根节点的二,若把根节点删去,则可以生成若干棵一根节点的儿子为根结点的子树,对于这些树来说,其上的路径也可以分为两种:1.经过根节点的 ...
[洛谷P4721]【模板】分治 FFT
题目大意:给定长度为$n-1$的数组$g_{[1,n)}$,求$f_{[0,n)}$,要求: $$ f_i=\sum_{j=1}^if_{i-j}g_j\\ f_0=1 $$ 题解:直接求复杂度是$O ...
Luogu 4721 【模板】分治 FFT
还不会这题的多项式求逆的算法. 发现每一项都是一个卷积的形式,那么我们可以使用$NTT$来加速,直接做是$O(n^2logn)$的,我们考虑如何加速转移. 可以采用$cdq$分治的思想,对于区间$[l ...
LuoGuP4721:【模板】分治 FFT
Pre 式子变换需要注意一下 Solution 注意到$f(x)g(x)+f_0=f(x)$ 其实开始我没看出来,后来发现仔细分析一下就可以了. 然后式子变换 \(f(x)=\frac{f_0}{ ...
凸包模板（分治 or Graham扫描法）
问题概述:空间上有很多点,现在要用一个凸多边形将所有点全部包住,求哪些点在这个凸多边形上输入样例: 对应输出: ...
点分治题单(来自XZY)
点分治题单(来自XZY) 静态点分治 [x] 洛谷 P3806 [模板]点分治1 [x] 洛谷 P4178 Tree [x] 洛谷 P2634 [国家集训队]聪聪可可 [x] 洛谷 P4149 [IO ...
【学习笔记】分治FFT
整理的算法模板合集: ACM模板点我看算法全家桶系列!!! 实际上是一个全新的精炼模板整合计划目录分治FFT 1. Luogu P4721 [模板]分治 FFT 2. 2020 ICPC Mac ...
FTT NTT 分治FFT
FFT study from: http://www.orchidany.cf/2019/02/19/FFT1/ https://www.cnblogs.com/zwfymqz/p/8244902.h ...

P3806 【模板】点分治1

P3806 【模板】点分治1相关推荐

最新文章

热门文章