首页 > 其他分享 >lca总结+树上差分

lca总结+树上差分

时间:2024-07-30 15:08:23浏览次数:12  
标签:return int sum 差分 -- num maxn lca 树上

lca

lca简称最近公共祖先——简介在此,不过多赘述

这里主要写的是倍增算法,oi-wiki上用的是vector,由于本人不会,只会用链表,所以这里就放链表的代码了

例题

image
image

加一个数组按倍增数组的方式存距离即可

题解——点击查看代码
#include<bits/stdc++.h>
#define int long long
const int maxn=1e6+10;
using namespace std;
int n,m,root,nxt[maxn<<2],to[maxn<<2],head[maxn<<2],tot,val[maxn<<2];
int cnt,dep[maxn<<2],f[maxn][20],dis[maxn][20];
void add(int x,int y,int z)
{
	to[++tot]=y;
	val[tot]=z;
	nxt[tot]=head[x];
	head[x]=tot;
}
void dfs(int u,int fa,int dist)
{
	dis[u][0]=dist;
	f[u][0]=fa;
	dep[u]=dep[fa]+1;
	for(int i=1;(1<<i)<=dep[u];i++)
	{
		f[u][i]=f[f[u][i-1]][i-1];
		dis[u][i]=dis[u][i-1]+dis[f[u][i-1]][i-1];
	}	
	for(int i=head[u];i;i=nxt[i])
	{
		int y=to[i];
		if(y==fa) continue;
		dfs(y,u,val[i]);
	}
}
int lca(int x,int y)
{
	int res=0;
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=17;i>=0;i--)
	{
		if(dep[y]+(1<<i)<=dep[x]) res+=dis[x][i],x=f[x][i];	
	}
	if(x==y) return res;
	for(int i=17;i>=0;i--)
	{
		if(f[y][i]!=f[x][i])
		{
			res+=dis[x][i];
			res+=dis[y][i];
			x=f[x][i];
			y=f[y][i];
		}
	}
	return dis[x][0]+dis[y][0]+res;
}

signed main()
{
	scanf("%d%d",&n,&m);
	char aa[2];
	int x,y,z;
	for(int i=1;i<=m;i++)
	{
		scanf("%lld%lld%lld %s ",&x,&y,&z,&aa[1]);
		add(x,y,z);
		add(y,x,z);
	} 
	dfs(1,0,0);
	int k;
	scanf("%lld",&k); 
	for(int i=1;i<=k;i++)
	{
		scanf("%lld%lld",&x,&y);
		printf("%lld\n",lca(x,y));
	}

	return 0;
}

树上差分

主要用途是在树上的一些统计计数操作,对树上两点路径的操作用的

主要思想

dfs深搜实现,在两端结点计数,在深搜回溯时把计数的操作从子节点传递到父节点,用下面这个图理解一下

image

假设我们有这样一张图,我们要在6和8两个节点间的路径加一,那我们在用树上差分时,在两端+1,会发现它们的最近公共

祖先2及以上祖先都加了2,我们的目的是让2加1,2的祖先不变,则我们需要在2处减一,2的父亲处再减一即可

[JLOI2014] 松鼠的新家

题解
#include<bits/stdc++.h>
#define int long long
const int maxn=1e6+10;
using namespace std;
int n,m,root,head[maxn<<2],tot,sum;
int cnt,dep[maxn<<2],f[maxn][21],tg[maxn],a[maxn];
bool vis[maxn<<2];
struct tree{int to,val,nxt;}e[maxn<<2];
void add(int x,int y)
{
	e[++tot].to=y;
	e[tot].nxt=head[x];
	head[x]=tot;
}

void dfs1(int u,int fa)
{
	vis[u]=1;
	for(int i=1;(1<<i)<=dep[u];i++)
	{
		f[u][i]=f[f[u][i-1]][i-1];
	}
	for(int i=head[u];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(vis[y]||y==fa) continue;
		f[y][0]=u;
		dep[y]=dep[u]+1;
		dfs1(y,u);
	}
}

int lca(int x,int y)
{
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20;i>=0;i--)
	{
		if(dep[x]>=dep[y]+(1<<i)) x=f[x][i];
	}
	if(x==y) return x;
	for(int i=20;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

void dfs(int u)
{
	vis[u]=1;
	for(int i=head[u];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(vis[y])continue;
		dfs(y);
		tg[u]+=tg[y];
	}
}

signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
	} 
	int x,y;
	for(int i=1;i<n;i++)
	{
		scanf("%lld%lld",&x,&y);
		add(x,y);
		add(y,x); 
	} 
	dfs1(1,0);
	for(int i=1;i<n;i++)
	{
		tg[a[i]]++;
		tg[a[i+1]]++;
		tg[ lca(a[i],a[i+1]) ]--;
		tg[f[ lca(a[i],a[i+1]) ][0]]--;
		/*
		en~~,神奇的东西,相邻的加1,但从a[2]到a[3]时a[2]不用加1,这就导致除了首尾糖果都 多加了1,但到最后一个糖果时不用拿,所以只有第一个位置的糖果是对的
		所以让第一个糖果数加1,最后输出时都减一就好了 
		*/ 
	}
	memset(vis,0,sizeof vis);
	dfs(1);
	tg[a[1]]++; 
	for(int i=1;i<=n;i++) printf("%lld\n",tg[i]-1);

	return 0;
}


拓展

1.求树上一点走k步可以走到的点的个数

将其 \(k\) 级祖先内的点加 \(1\) 用树上差分可以实现,这里再加一种更简便的思路

\(dfs\) 一遍,维护一个栈,进一个点时记一下数量,回溯时再记一下数量,回溯是把距离超过 \(k\) 的点从栈中删了

这里记的两个数量差就是 \(k\) 步可以走到的。复杂度 \(O(n)\)

点击查看代码
void dfs(int &num,int x,int fa,int d)
{
	int t=num;
	if(d>top)st[++top]=0;
	num++,st[d]++;
	for(int i=head[x];i;i=nxt[i])
	{
		int y=to[i];
		if(y==fa)continue;
		dfs(num,y,x,d+1);
	}
	ans[x]=num-t;
	if(top==d+k) num-=st[top--];
}

2.环上差分

环上差分需要先找环,适用于环上的点对一段区间有影响问题

floyd找环法
while(u!=y&&u!=to[y])u=to[u],y=to[to[y]];

环上差分和树上差分差不多,都是信息一步一步向上传来统计的

[ICPC2019 WF] Hobson's Trains

题解
#include<bits/stdc++.h>
const int maxn=5e5+10;
using namespace std;
int n,k,tot,head[maxn],to[maxn],nxt[maxn],v[maxn];
int st[maxn],top,ans[maxn],a[maxn],cnt,s[maxn];

void add(int x,int y)
{
	to[++tot]=y;
	nxt[tot]=head[x];
	head[x]=tot;
} 

void dfs(int &num,int x,int fa,int d)
{
//	cout<<x<<" "<<fa<<" "<<d<<endl;
	int t=num;
	if(d>top)st[++top]=0;
	num++,st[d]++;
	for(int i=head[x];i;i=nxt[i])
	{
		int y=to[i];
		if(y==fa)continue;
		dfs(num,y,x,d+1);
	}
	ans[x]=num-t;
	if(top==d+k) num-=st[top--];
}
inline int mo(int i,int x){return i+x<=cnt?i+x:i+x-cnt;}
void solve(int x)
{
	int u=x,y=v[x];
	while(u!=y&&u!=v[y])u=v[u],y=v[v[y]];
	a[cnt=1]=u;
	for(int i=v[u];i!=u;i=v[i])a[++cnt]=i;
//	for(int i=1;i<=cnt;i++)s[i]=0;
	fill(s+1,s+cnt+1,0);
	int sum=min(k+1,cnt);
//	cout<<cnt<<" !"<<endl;
	for(int i=1;i<=cnt;i++)
	{
		for(int j=head[a[i]];j;j=nxt[j])
		{
			int y=to[j];
			if(y==a[mo(i,cnt-1)])continue;
			int num=0;
			st[top=0]=0;
			dfs(num,y,a[i],1);
			for(int d=1;d<=top;d++)
			{
				int temp=st[d];
				if(k-d>=cnt)sum+=temp;
				else
				{
					int m=mo(i,k-d+1);
					s[i]+=temp,s[m]-=temp;
					if(m<=i) s[1]+=temp;
				}
			} 
		}
	}
	for(int i=1;i<=cnt;i++) sum+=s[i],ans[a[i]]=sum;
}

int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0),cout.tie(0); 
	cin>>n>>k;
	for(int i=1;i<=n;i++)
	{
		int x;
		cin>>x;
		add(x,i);
		v[i]=x;
//		ans[i]=-1;
	}
	fill(ans+1,ans+1+n,-1);
	for(int i=1;i<=n;i++) 
		if(ans[i]==-1) solve(i);
	for(int i=1;i<=n;i++)
		cout<<ans[i]<<'\n';
	
	return 0;
}
/*
6 2
2 
3
4
5
4
3
*/

你已完成新手教程,下面开启困难模式

树上差分计数变形

[NOIP2015 提高组] 运输计划

题解——二分加树上差分
#include<bits/stdc++.h>
const int maxn=1e6+10;
using namespace std;
int n,m,root,head[maxn],tot,a[maxn],b[maxn];
int cnt,dep[maxn],f[maxn][21],dis[maxn][21],sum[maxn],l,r,ans;
bool vis[maxn];
struct tree{int to,val,nxt;}e[maxn<<2];
struct node{int a,b,anc,val;}le[maxn<<2]; 

void add(int x,int y,int z)
{
	e[++tot].to=y;
	e[tot].nxt=head[x];
	e[tot].val=z;
	head[x]=tot;
}

void dfs(int u,int fa)
{
	vis[u]=1;
	for(int i=1;(1<<i)<=dep[u];i++)
	{
		f[u][i]=f[f[u][i-1]][i-1]; 
		dis[u][i]=dis[u][i-1]+dis[f[u][i-1]][i-1];
	}
	for(int i=head[u];i;i=e[i].nxt)
	{
		int y=e[i].to;
		if(vis[y]||y==fa) continue;
		f[y][0]=u;
		b[y]=i;
		dis[y][0]=e[i].val;
		dep[y]=dep[u]+1;
		dfs(y,u);
	}
}

int lca(int x,int y)
{
	if(dep[x]>dep[y]) swap(x,y);
	for(int i=20;i>=0;i--)
	{
		if(dep[x]+(1<<i)<=dep[y]) y=f[y][i];	
	}
	if(x==y) return y;
	for(int i=20;i>=0;i--)
	{
		if(f[y][i]!=f[x][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[y][0];
}


int solve(int x,int y)
{
	int res=0;
	if(dep[x]<dep[y])swap(x,y);
	for(int i=17;i>=0;i--)
	{
		if(dep[x]>=dep[y]+(1<<i)) res+=dis[x][i],x=f[x][i];
	}
	if(x==y)return res;
	for(int i=17;i>=0;i--)
	{
		if(f[y][i]!=f[x][i])
		{
			res+=dis[x][i]+dis[y][i];
			x=f[x][i];
			y=f[y][i];
		}
	}
	return dis[x][0]+dis[y][0]+res;
}

void update(int now,int fa)
{
	for(int i=head[now];i;i=e[i].nxt)
	{
		if(e[i].to!=fa)
		{
			update(e[i].to,now);
			sum[now]+=sum[e[i].to];
		}
	}
}

bool check(int x)
{
	int cnt=0,dec=0;
	memset(sum,0,sizeof sum);
	for(int i=1;i<=n;i++)
	{
		if(le[i].val>x)
		{
			cnt++;
			sum[le[i].a]++;
			sum[le[i].b]++;
			sum[le[i].anc]-=2;
			dec=max(dec,le[i].val-x);
		}
	}
	update(1,1);
	for(int i=1;i<=n;i++)
		if(sum[i]==cnt&&e[b[i]].val>=dec) return 1;
		return 0; 
}

int main()
{
	scanf("%d%d",&n,&m);
	int x,y,z;
	for(int i=1;i<n;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		add(x,y,z);
		add(y,x,z);
	}
	dfs(1,0);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		le[i]={x,y,lca(x,y),solve(x,y)};
		r=max(r,le[i].val);
	} 
	r++;
	while(l<r)
	{
		int mid=(l+r)>>1;
		if(check(mid))ans=r=mid;
		else l=mid+1;
	}
	printf("%d",ans);

	return 0;
}

树上差分加线段树合并

[Vani有约会] 雨天的尾巴

题解
#include<bits/stdc++.h>
#define lid m[id].ls
#define rid m[id].rs
const int maxn=1e5+10;
using namespace std;
int n,t,len,head[maxn],nxt[maxn<<1],to[maxn<<1],tot,cnt;
int rt[maxn],sum,s[maxn*80],ans[maxn];
struct node{int ls,rs,sum;}m[maxn*80];
int dep[maxn],f[maxn][21];

inline void add(int x,int y)
{
	to[++cnt]=y;
	nxt[cnt]=head[x];
	head[x]=cnt;
}

inline void addm(int x,int y)
{
	add(x,y);
	add(y,x);
}

inline void push(int id)
{
	if(!lid){m[id].sum=m[rid].sum,s[id]=s[rid];return ;}
	if(!rid){m[id].sum=m[lid].sum,s[id]=s[lid];return ;} 
	m[id].sum=max(m[lid].sum,m[rid].sum);
	s[id]=m[lid].sum>=m[rid].sum?s[lid]:s[rid]; 
}

inline void merge(int &a,int b,int l,int r)
{
    if(!b) return;
    if(!a){ a=b;return ;}
    if(l==r){ m[a].sum+=m[b].sum;return ;}
    int mid=(l+r)>>1;
    merge(m[a].ls,m[b].ls,l,mid),merge(m[a].rs,m[b].rs,mid+1,r);
    push(a);
}

inline void insert(int &id,int l,int r,int x,int y)
{
	if(!id)id=++tot;
	if(l==r)
	{
		m[id].sum+=y;
		s[id]=x;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)insert(lid,l,mid,x,y);
	else insert(rid,mid+1,r,x,y);
	push(id);
} 

inline void dfs(int u,int fa)
{
	for(int i=1;(1<<i)<=dep[u];i++)
		f[u][i]=f[f[u][i-1]][i-1];
	for(int i=head[u];i;i=nxt[i])
	{
		int y=to[i];
		if(y==fa)continue;
		f[y][0]=u;
		dep[y]=dep[u]+1;
		dfs(y,u);
	}
} 

inline int lca(int x,int y)
{
	if(dep[x]<dep[y])swap(x,y);
	for(int i=20;i>=0;i--)
		if(dep[y]+(1<<i)<=dep[x])x=f[x][i];
	if(x==y)return x;
	for(int i=20;i>=0;i--)
	{
		if(f[y][i]!=f[x][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

void as(int u,int fa)
{
	for(int i=head[u];i;i=nxt[i])
	{
		int y=to[i];
		if(y==fa)continue;
		as(y,u);
		merge(rt[u],rt[y],1,maxn-10);
	}
	ans[u]=s[rt[u]];
	if(!m[rt[u]].sum) ans[u]=0;
}

int main(){
	scanf("%d%d",&n,&t);
	for(int i=1;i<n;i++)
	{
		int x,y;
		scanf("%d%d",&x,&y);
		addm(x,y);
	} 
	dfs(1,0);
	for(int i=1;i<=t;i++)
	{
		int a,b,c,d; 
		scanf("%d%d%d",&a,&b,&c);
		d=lca(a,b);
		insert(rt[a],1,maxn,c,1),insert(rt[b],1,maxn,c,1);
		insert(rt[d],1,maxn,c,-1);
		insert(rt[f[d][0]],1,maxn,c,-1); 
	} 
	as(1,0);
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]);
    return 0;
}

标签:return,int,sum,差分,--,num,maxn,lca,树上
From: https://www.cnblogs.com/oceansofstars/p/18184517

相关文章

  • 洛谷题单指南-前缀和差分与离散化-P3017 [USACO11MAR] Brownie Slicing G
    原题链接:https://www.luogu.com.cn/problem/P3017题意解读:将一个r*c的矩阵,横向切成a条,每一条纵向切除b块,计算每一块子矩阵之和的最小值最大是多少。解题思路:要计算最小值中最大的,直觉上可以采用二分,下面来分析单调性:给定一个子矩阵块之和的值,值越小可以划分的条数、块数就越多......
  • 洛谷题单指南-前缀和差分与离散化-P2004 领地选择
    原题链接:https://www.luogu.com.cn/problem/P2004题意解读:在一个n*m的矩阵中,找到边长为c的正方形,使得正方形范围内的数之和最大,输出正方形的左上角坐标。解题思路:二维前缀和的简单应用第一步:计算二维前缀和第二步:枚举边长为c的正方形左上角,计算正方形区域的数之和,更新答案第......
  • 洛谷题单指南-前缀和差分与离散化-P1884 [USACO12FEB] Overplanting S
    原题链接:https://www.luogu.com.cn/problem/P1884题意解读:给定n个矩形的平面直角坐标系下左上角、右下角的坐标,计算这n个矩形能覆盖的的格子数。解题思路:直观上来看,此题是一个差分应用,针对二维差分数组,将n个矩形区域内每个格子的值加1,然后统计有多少个不为0的格子即可。但是!坐......
  • 洛谷题单指南-前缀和差分与离散化-P1496 火烧赤壁
    原题链接:https://www.luogu.com.cn/problem/P1496题意解读:给定n个区间[a,b),计算所有区间覆盖的总长度。解题思路:方法1、离散化先思考一种比较直观的思路:既然要计算多个区间覆盖的总长度,可以枚举每一个区间[a,b),通过一个桶数组来标记区间中所有的点f[x]=1,最终统计所有为1的......
  • 前缀和与差分
    前缀和与差分前缀和定义:前缀和可以简单理解为「数列的前n项的和」,是一种重要的预处理方式,能大大降低查询的时间复杂度。一维前缀和模板for(inti=1;i<=n;i++)sum[i]=sum[i-1]+a[i];时间复杂度:O(n)原理数组sum用于储存前i个元素的和,数组a......
  • Luogu P3177 树上染色 [ 蓝 ] [ 树形 dp ] [ 贡献思维 ]
    一道很好的树形dp!!!!!树上染色。错误思路定义\(dp[u][i]\)表示以\(u\)为根的子树中,把\(i\)个点染成黑色的最大收益。但这样写,就在转移的时候必须枚举每一个点,复杂度过大,而且还不好写,是十分错误的写法。正确思路一般看到有关树上“路径”的题,就要把路径拆成一个个独立的......
  • 差分约束
    差分约束差分约束是一种特殊的n元一次不等式组,m个约束条件,可以组成形如下的格式:\[\begin{cases}x_1-x_1^{'}\ley_1\\x_2-x_2^{'}\ley_2\\\cdots\\x_m-x_m^{'}\ley_m\end{cases}\]我们的任务是需要求出一组解,\(x_1=a_1,x_2=a_2,\cdots,x_n=a_n\)使得不等式......
  • LCA学习笔记
    LCA\(\textbf{LCA=LowestCommonAncestor}\)即最近公共祖先下文以\(\textbf{LCA(a,b)表示节点a与节点b的最近公共祖先}\)F1:暴力算法步骤:(1)求出每个节点的深度(\(size\))(2)询问两个点是否重合,若重合,则\(\textbf{LCA(a,b)=当前重合的节点}\)(3)否则,选择\(\textbf{max(size[a]......
  • 表面贴装差分输出晶体振荡器 DSO533SK/DSO533SJ 全面解析
    表面贴装差分输出晶体振荡器DSO533SK/DSO533SJ全面解析摘要:本文深入探讨表面贴装差分输出晶体振荡器DSO533SK/DSO533SJ的特性、工作原理、优势、应用领域以及使用中的关键要点。通过详细的阐述和实例分析,为读者提供全面且深入的技术解读。一、引言在当今高度集成和高......
  • 洛谷题单指南-前缀和差分与离散化-P3397 地毯
    原题链接:https://www.luogu.com.cn/problem/P3397题意解读:给定一个n*n的矩阵,每个元素初始值为0,再将m个子矩阵中的元素都增加1,统计每个元素最终的值。解题思路:1、暴力法枚举每一个子矩阵,将对应元素值加1,时间复杂度为1000^3,不可行。2、二维差分对于给定二维数组s[][],给指定区......