点分治模板+例题

最新推荐文章于 2022-04-27 12:52:47 发布

冰可乐小朋友

最新推荐文章于 2022-04-27 12:52:47 发布

阅读量218

点赞数

分类专栏：点分治

本文链接：https://blog.csdn.net/weixin_43819578/article/details/102144715

版权

点分治专栏收录该内容

1 篇文章 0 订阅

订阅专栏

1.找到树的重心，更新以重心为根节点的子树大小。
2.分治，从重心开始，dfs找每个点到根节点的距离，看是否存在题意路径。
3.把根节点删去，重复1 2 步骤
例题：
1.P3806 【模板】点分治1链接
给定一棵有n个点的树

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

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
using namespace std;
const int N=1e5+5;
const int maxn=1e7+5;
int n,m;
int tot,rt,sum,cnt;
int he[N],ne[N*2],e[N*2],ver[N*2],q[105],vis[N];
int siz[N],maxp[N],tmp[N],dis[N];
bool jd[maxn],ans[105];
void add(int x,int y,int w){
	ver[++tot]=y;
	ne[tot]=he[x];
	e[tot]=w;
	he[x]=tot;
}
void getrt(int u,int f){
	siz[u]=1;//子树大小
	maxp[u]=0;//u为根的最大子树(包括父亲树）
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(v==f||vis[v])continue;//vis true代表此节点被删 
		getrt(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxp[u])maxp[u]=siz[v];
	} 
	maxp[u]=max(maxp[u],sum-siz[u]);
	if(maxp[u]<maxp[rt])rt=u;
}
void getdis(int u,int f){
	tmp[cnt++]=dis[u];//把所有距离放进一个数组里
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(v==f||vis[v])continue;
		dis[v]=dis[u]+e[i];
		getdis(v,u);
	}
}
void solve(int u){
	queue<int> que;//队列用于对jd的清空 
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(vis[v])continue;
		cnt=0;
		dis[v]=e[i];
		getdis(v,u);
		for(int j=0;j<cnt;j++){//遍历所有的长度 
			for(int k=0;k<m;k++){//遍历所有的询问 
				if(q[k]>=tmp[j])//有可能存在
				{
					ans[k]|=jd[q[k]-tmp[j]]; 
				} 
			}
		}
		for(int j=0;j<cnt;j++){
			que.push(tmp[j]);
			jd[tmp[j]]=1; 
		}
	} 
	while(que.size()){//不能用memset 
		jd[que.front()]=false;
		que.pop();
	}
}
void divide(int u){
	vis[u]=jd[0]=1;//删u节点（根节点），jd表示某路径（长度）在之前的子树是否存在
	solve(u);//处理所有经过u的路径
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(vis[v])continue;
		maxp[rt=0]=sum=siz[v];//rt初始化为0，即maxp[rt]为极大值 
		getrt(v,0);
		getrt(rt,0);
		divide(rt);
	} 
	 
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<n;i++){
		int x,y,w;
		scanf("%d%d%d",&x,&y,&w);
		add(x,y,w);
		add(y,x,w);
	}
	for(int i=0;i<m;i++){
		scanf("%d",&q[i]);
	}
	maxp[0]=sum=n;
	getrt(1,0);
	getrt(rt,0); //以rt为根再算一遍siz[v]
	divide(rt); 
	for(int i=0;i<m;i++){
		if(ans[i])puts("AYE");
		else puts("NAY");
	}
	return 0;
}

2.CF161D Distance in Tree

输入点数为N一棵树
求树上长度恰好为K的路径个数

这里的jd数组不仅仅是看它有没有，这里用作桶数组，看之前的子树与当前匹配有多少种情况。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=5e4+5;
int n,k,tot,ans,rt,sum,cnt;
int siz[maxn],maxp[maxn],tmp[maxn],dis[maxn],vis[maxn];
int jd[maxn*2];
struct node{
	int to,ne,w;
}edge[maxn*2];
int fir[maxn];
void add(int u,int v,int w){
	edge[++tot].to=v;
	edge[tot].w=w;
	edge[tot].ne=fir[u];
	fir[u]=tot;
}
void getrt(int u,int f){
	siz[u]=1;
	maxp[u]=0;
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(v==f||vis[v])continue;
		getrt(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxp[u])maxp[u]=siz[v];
	}
	maxp[u]=max(maxp[u],sum-siz[u]);
	if(maxp[u]<maxp[rt])rt=u;
}
void getdis(int u,int f){
	tmp[cnt++]=dis[u];
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(v==f||vis[v])continue;
		dis[v]=dis[u]+edge[i].w;
		getdis(v,u);
	}
}
void solve(int u){
	queue<int> que;
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(vis[v])continue;
		cnt=0;
		dis[v]=edge[i].w;
		getdis(v,u);
		for(int j=0;j<cnt;j++){
			if(k>=tmp[j]){
				ans+=jd[k-tmp[j]];
			}
		}
		for(int j=0;j<cnt;j++){
			que.push(tmp[j]);
			jd[tmp[j]]++; 
		}
	}
	while(que.size()){
		jd[que.front()]=0;
		que.pop();
	}
}
void divide(int u){
	vis[u]=jd[0]=1;
	solve(u);
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(vis[v])continue;
		maxp[rt=0]=sum=siz[v];
		getrt(v,0);
		getrt(rt,0);
		divide(rt);
	}
}

int main(){
	scanf("%d%d",&n,&k);
	int x,y,w;
	tot=0;
	for(int i=1;i<n;i++){
		scanf("%d%d",&x,&y);
		add(x,y,1);
		add(y,x,1);
	}
	maxp[0]=sum=n;
	getrt(1,0);
	getrt(rt,0);
	divide(rt);
	printf("%d\n",ans);
	return 0;
}

3.洛谷 P4178 Tree
给你一棵TREE,以及这棵树上边的距离.问有多少对点它们两者间的距离小于等于K。
需要树状数组维护
change（x,y)表示在x的位置上加上y；
ask（n）表示小于n的数一共有几个，都是板子，因为树状数组为0时可能会有问题，所以这里维护时要+1.

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=1e5+5;
int n,k,tot,ans,rt,sum,cnt;
int siz[maxn],maxp[maxn],tmp[maxn],dis[maxn],vis[maxn];
int jd[maxn*100];
struct node{
	int to,ne,w;
}edge[maxn*2];
int fir[maxn];
void add(int u,int v,int w){
	edge[++tot].to=v;
	edge[tot].w=w;
	edge[tot].ne=fir[u];
	fir[u]=tot;
}
void change(int x,int y){
	for(;x<=100*maxn-2;x+=x&(-x)){
		jd[x]+=y;
	}
}
long long ask(int x){
	long long ans=0;
	for(;x;x-=x&(-x)){
		ans+=jd[x];
	}
	return ans;
}
void getrt(int u,int f){
	siz[u]=1;
	maxp[u]=0;
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(v==f||vis[v])continue;
		getrt(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxp[u])maxp[u]=siz[v];
	}
	maxp[u]=max(maxp[u],sum-siz[u]);
	if(maxp[u]<maxp[rt])rt=u;
}
void getdis(int u,int f){
	tmp[cnt++]=dis[u];
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(v==f||vis[v])continue;
		dis[v]=dis[u]+edge[i].w;
		getdis(v,u);
	}
}
void solve(int u){
	queue<int> que;
	que.push(0);
	change(1,1); 
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(vis[v])continue;
		cnt=0;
		dis[v]=edge[i].w;
		getdis(v,u);
		for(int j=0;j<cnt;j++){
			if(k>=tmp[j]){
				ans+=ask(k-tmp[j]+1);
			}
		}
		for(int j=0;j<cnt;j++){
			que.push(tmp[j]);
			change(tmp[j]+1,1);
		}
	}
	while(que.size()){
		change(que.front()+1,-1);
		que.pop();
	}
}
void divide(int u){
	vis[u]=jd[0]=1;
	solve(u);
	for(int i=fir[u];i;i=edge[i].ne){
		int v=edge[i].to;
		if(vis[v])continue;
		maxp[rt=0]=sum=siz[v];
		getrt(v,0);
		getrt(rt,0);
		divide(rt);
	}
}

int main(){
	scanf("%d",&n);
	int x,y,w;
	tot=0;
	for(int i=1;i<n;i++){
		scanf("%d%d%d",&x,&y,&w);
		add(x,y,w);
		add(y,x,w);
	}
	scanf("%d",&k);
	maxp[0]=sum=n;
	getrt(1,0);
	getrt(rt,0);
	divide(rt);
	printf("%d\n",ans);
	return 0;
}

4.P2634 [国家集训队]聪聪可可
求两点之间的距离是3的倍数的点对有多少个
注意放进jd时都要mod 3，不然可能会数组越界。

#include<iostream>
#include<cstdio>
#include<queue>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=20005;
const int maxn=1e7+5;
int n,m;
int tot,rt,sum,cnt;
int he[N],ne[N*2],e[N*2],ver[N*2],q[105],vis[N];
int siz[N],maxp[N],tmp[N],dis[N];
int jd[maxn],ans;
void add(int x,int y,int w){
	ver[++tot]=y;
	ne[tot]=he[x];
	e[tot]=w;
	he[x]=tot;
}
void getrt(int u,int f){
	siz[u]=1;//子树大小
	maxp[u]=0;//u为根的最大子树(包括父亲树）
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(v==f||vis[v])continue;//vis true代表此节点被删 
		getrt(v,u);
		siz[u]+=siz[v];
		if(siz[v]>maxp[u])maxp[u]=siz[v];
	} 
	maxp[u]=max(maxp[u],sum-siz[u]);
	if(maxp[u]<maxp[rt])rt=u;
}
void getdis(int u,int f){
	tmp[cnt++]=dis[u];//把所有距离放进一个数组里
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(v==f||vis[v])continue;
		dis[v]=dis[u]+e[i];
		getdis(v,u);
	}
}
void solve(int u){
	queue<int> que;//队列用于对jd的清空 
	que.push(0); 
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(vis[v])continue;
		cnt=0;
		dis[v]=e[i];
		getdis(v,u);
		for(int j=0;j<cnt;j++){//遍历所有的长度 
			if(tmp[j]%3==0)ans+=jd[0];
			else if(tmp[j]%3==1)ans+=jd[2];
			else ans+=jd[1];
		}
		for(int j=0;j<cnt;j++){
			que.push(tmp[j]);
			jd[tmp[j]%3]++;
		}
	} 
	while(que.size()){//不能用memset 
		jd[que.front()%3]--;
		que.pop();
	}
}
void divide(int u){
	vis[u]=jd[0]=1;//删u节点（根节点），jd表示某路径（长度）在之前的子树是否存在
	solve(u);//处理所有经过u的路径
	for(int i=he[u];i;i=ne[i]){
		int v=ver[i];
		if(vis[v])continue;
		maxp[rt=0]=sum=siz[v];//rt初始化为0，即maxp[rt]为极大值 
		getrt(v,0);
		getrt(rt,0);
		divide(rt);
	} 
	 
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int x,y,w;
		scanf("%d%d%d",&x,&y,&w);
		add(x,y,w);
		add(y,x,w);
	}
	maxp[0]=sum=n;
	getrt(1,0);
	getrt(rt,0); //以rt为根再算一遍siz[v]
	divide(rt); 
	ans=ans*2+n;
	int ff=n*n;
	int g=__gcd(ans,ff);
	printf("%d/%d",ans/g,ff/g);
	return 0;
}