倍增 ST表 & RMQ & LCA

本文通过三个实例展示了数据结构与算法在解决信息技术问题中的重要作用。第一部分介绍了LCA最近公共祖先的计算,利用链式前向星进行图的构建;第二部分讨论了如何利用并查集和LCA计算两点间的距离;第三部分涉及二维ST表在区间最值问题上的应用。这些案例揭示了数据结构和算法在高效解决复杂问题中的核心价值。
摘要由CSDN通过智能技术生成

1. LCA 最近公共祖先

板子(链式前向星建图):

void dfs(int x,int fa)
{
	for(int i=fi[x];i!=-1;i=nxt[i])
	{
		int k=e[i].to;
		if(k==fa)
		  continue;
	    dep[k]=dep[x]+1;
	    f[k][0]=x;
	    for(int j=0;j<=17;j++)
	    {
	    	f[k][j]=f[f[k][j-1]][j-1];
		}
		dfs(k,x);
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])
	  return lca(y,x);
	for(int i=17;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y])
		{
			x=f[x][i];
		}
	}
	if(x==y)
	  return x;
	for(int i=17;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}

练习
1.poj1330
思路:板子题。

#include<bits/stdc++.h>
using namespace std;
const int N=30001;
struct node
{
	int nxt,to;
}e[N];
int pre[N];
int n;
int fi[N];
int f[N][20],dep[N];
int cnt=0;
int t;
void add(int x,int y)
{
	e[++cnt].to=y;
	e[cnt].nxt=fi[x];
	fi[x]=cnt;
	e[++cnt].to=x;
	e[cnt].nxt=fi[y];
	fi[y]=cnt;
}


void dfs(int x,int fa)
{
	for(int i=fi[x];i!=-1;i=e[i].nxt)
	{
		int j=e[i].to;
		if(j==fa)
		  continue;
		dep[j]=dep[x]+1;
		f[j][0]=x;
		for(int i=1;i<=17;i++)
		{
			f[j][i]=f[f[j][i-1]][i-1];
		}
		dfs(j,x);
	}
}


int lca(int x,int y)
{
	if(dep[y]>dep[x])
	{
		return lca(y,x);
	}
	for(int i=17;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y])
		  x=f[x][i];
	}
	if(x==y)
	  return x;
	for(int i=17;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
void solve()
{
	memset(fi,-1,sizeof(fi));
	memset(f,0,sizeof(f));
	memset(dep,0,sizeof(dep));
	memset(pre,0,sizeof(pre));
	cnt=0;
	cin>>n;
	int x,y;
	int root;
	for(int i=1;i<n;i++)
	{
		
		cin>>x>>y;
		pre[y]++;
		add(x,y);
	}
	for(int i=1;i<=n;i++)
	{
		if(pre[i]==0)
		{
			root=i;
			break;
		}
	}
	dep[root]=1;
    dfs(root,-1);
	cin>>x>>y;
	cout<<lca(x,y)<<endl;
}
int main()
{
	cin>>t;
	while(t--)
	{
		solve();
	}
	return 0;
}

2.HUD2586
思路:A,B 2个节点的距离 == A到根的距离+B到根的距离-2*lca(A,B)的距离

#include<bits/stdc++.h>
using namespace std;
int t;
const int N=4e4+11;
int n,m;
struct node
{
	int nxt,to,w;
}e[N*2];
int first[N];
int f[N][20];
int dep[N];
int pre[N];
int dis[N];
int cnt=0;
void dfs(int x,int fa)
{
	for(int i=first[x];i!=-1;i=e[i].nxt)
	{
		int j=e[i].to;
		if(j==fa)
		  continue;
		f[j][0]=x;
		for(int i=1;i<=17;i++)
		{
			f[j][i]=f[f[j][i-1]][i-1];
		}
		dep[j]=dep[x]+1;
		dis[j]=dis[x]+e[i].w;
		dfs(j,x);
	}
}
int lca(int x,int y)
{
	if(dep[x]<dep[y])
	  return lca(y,x);
	for(int i=17;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y])
		{
			x=f[x][i];
		}
	}
	if(x==y)
	  return x;
	for(int i=17;i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
void init()
{
	cnt =0;
	memset(first,-1,sizeof(first));
	memset(f,0,sizeof(f));
	memset(dep,0,sizeof(dep));
	memset(dis,0,sizeof(dis));
	memset(pre,0,sizeof(pre));
}
void add(int x,int y,int z)
{
	e[++cnt].to=y;
	e[cnt].nxt=first[x];
	first[x]=cnt;
	e[cnt].w=z;	
}
void solve()
{
	int root;
	for(int i=1;i<=n;i++)
	{
		if(pre[i]==0)
		{
			root=i;
			break;
		}
	}
	dep[root]=1;
	dis[root]=0;
	
	dfs(root,-1);
	while(m--)
	{
		int x, y;
		cin>>x>>y;
		cout<<dis[x]+dis[y]-2*dis[lca(x,y)]<<endl;
	}
}
int main()
{
	cin>>t;
	while(t--)
	{
		cin>>n>>m;
		init();
		int x,y,z;
		for(int i=1;i<n;i++)
		{
			cin>>x>>y>>z;
			add(x,y,z);
			add(y,x,z);
			pre[y]++;
		}
		solve();	
	}	
 } 

3.HUD2874
思路:并查集+第二题的思路

#include<bits/stdc++.h>
using namespace std;
int n,m,c;
const int N=1e5+11;
int fa[N];
struct node
{
	int w;
	int nxt,to;
}e[2*N];
int h[N*2];
int cnt;
int pre[N];
int dis[N];
int dep[N];
int f[N][20];
int root;
int find(int x)
{
	if(x==fa[x])
	  return x;
	fa[x]=find(fa[x]);
	return fa[x];	
}
void Add(int x,int y)
{
	x=find(x);
	y=find(y);
	if(x!=y)
	  fa[y]=x;
}
void add(int x,int y,int z)
{
	e[++cnt].w=z;
	e[cnt].to=y;
	e[cnt].nxt=h[x];
	h[x]=cnt;
}
int lca(int x,int y)
{
	if(dep[y]>dep[x])
	  return lca(y,x);
	for(int i=17;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y])
		{
			x=f[x][i];
		}
	}
	if(x==y)
	  return x;
	for(int i=17;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 x,int fa)
{
	for(int i=h[x];i!=-1;i=e[i].nxt)
	{
		int j=e[i].to;
		if(j==fa)
		  continue;
		for(int i=1;i<=17;i++)
		{
			f[j][i]=f[f[j][i-1]][i-1];
		}
		dep[j]=dep[x]+1;
		dis[j]=dis[x]+e[i].w;
		dfs(j,x);	
	}
     	
}
void init()
{
	for(int i=1;i<=n;i++)
	{
		fa[i]=i;
	}
	memset(dis,0,sizeof(dis));
	memset(f,0,sizeof(f));
	memset(pre,0,sizeof(pre));
	memset(dep,0,sizeof(dep));
	memset(h,-1,sizeof(h));
	cnt=0;
}
void solve()
{
	int x,y,z;
	for(int i=1;i<=m;i++)
	{
		cin>>x>>y>>z;
		add(x,y,z);
		add(y,x,z);
		pre[y]++;
		Add(x,y);
	}
	for(int i=1;i<=n;i++)
	{
		if(pre[i]==0)
		{
			dep[i]=1;
			dis[i]=0;
			dfs(i,-1);
		}
	}
	while(c--)
	{
		cin>>x>>y;
		if(find(x)!=find(y))
		{
			cout<<"Not connected"<<endl;
		}
		else
		{
			cout<<dis[y]+dis[x]-2*dis[lca(x,y)]<<endl;
		}
	}
}
int main()
{
	while(cin>>n>>m>>c)
	{
		init();
		solve();
	}
	return 0;
}

2. RMQ区间最值问题

ST表

void ST()
{
	for(int i=1;i<=n;i++)
	  f[i][0]=a[i];
	int k=log(n);
	for(int i=1;i<=k;i++)
	{
		for(int j=1;j+(1<<i)<=n+1;j++)
		{
			f[j][i]=max(f[j][i-1],f[j+(1<<(i-1))][i-1])
		}
	}	
}
int ST_find(int l,int r)
{
	int k=log2((r-l+1));
	return max(f[l][k],f[r-(1<<k)+1][k]); 
}

1.poj3264
思路:f1存区间最大,f2存区间最小

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
int n,q;
const int N=5e5+11;
int a[N];
int f1[N][20];
int f2[N][20];
void ST()
{
	for(int i=1;i<=n;i++)
	{
		f1[i][0]=a[i];
		f2[i][0]=a[i];
	}
	   int k=log(n)/log(2);
       for (int j = 1; (1 << j) <= n; j ++ )
       {
	   
		for (int i = 1; i + (1 << j) - 1 <= n; i ++ )
		{
		
			f1[i][j] = max(f1[i][j - 1], f1[i + (1 << (j - 1))][j - 1]);
			f2[i][j]=min(f2[i][j-1],f2[i+(1<<(j-1))][j-1]);
		 
		}
	}	
}
int ST_find1(int l,int r)
{
	int k=log(r-l+1)/log(2);
	return max(f1[l][k],f1[r-(1<<k)+1][k]);
}
int ST_find2(int l,int r)
{
	int k=log(r-l+1)/log(2);
	return min(f2[l][k],f2[r-(1<<k)+1][k]);
}

int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>q;
	for(int i=1;i<=n;i++)
	{
	    cin>>a[i];
	}
//	memset(f2,200001,sizeof(f2));
	ST();
	int l,r;
	while(q--)
	{
		cin>>l>>r;
		cout<<ST_find1(l,r)-ST_find2(l,r)<<endl;
	}
	return 0; 	  
}

2.poj 3368
思路:…ez

#include<iostream>
#include<cmath>
using namespace std;
int n,m;
const int N=1e5+11;
int e[N];
int f[N][20];
int a[N];
int find(int l,int r)
{
	int k=log(r-l+1)/log(2);
	return max(f[l][k],f[r-(1<<k)+1][k]);	
}
void st()
{
	int k=log(n)/log(2);
	for(int i=1;i<=n;i++)
	{
		f[i][0]=e[i];
	}
	for(int i=1;i<=k;i++)
	{
		for(int j=1;j+(1<<i)-1<=n;j++)
		{
			f[j][i]=max(f[j][i-1],f[j+(1<<(i-1))][i-1]);
		}
	}
}
void solve()
{
	int x;
   for(int i=1;i<=n;i++)
   	 cin>>a[i];
    for(int i=1;i<=n;i++)
    {
    	if(i==1)
    	  e[i]=1;
    	else
    	{
    		if(a[i]==a[i-1])
    		{
    			e[i]=e[i-1]+1;
			}
			else
			{
				e[i]=1;
			}
		}
	}
	st();
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	while(cin>>n && n)
	{
	      cin>>m;
	      solve();
	      int l,r;
		while(m--)
		{
		   cin>>l>>r;   
	       int t=0;
	       while(e[l]!=1 && l<=r)
	       {
	       	 l++;
	       	 t++;
		   }
		   if(l<=r)
		     t=max(t,find(l,r));
		   cout<<t<<endl;
		}		
	}	
}

3.poj2019(二维ST)
思路:将2维转为一维的

#include<iostream>
#include<cmath>
using namespace std;
int n,b,k;
const int N=257;
int a[N][N];
int f1[N][N][8][8];
int f2[N][N][8][8];
void st()
{
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		{
			f1[i][j][0][0]=a[i][j];
			f2[i][j][0][0]=a[i][j];
		}
	}
	int kk= log(b)/log(2);
	for(int k1=0;k1<=kk;k1++)
	{
		for(int k2=0;k2<=kk;k2++)
		{
			if(!k1 &&  !k2)
			  continue;
			for(int i=0;i+(1<<k1)-1<=n;i++)
			{
				for(int j=0;j+(1<<k2)-1<=n;j++)
				{
					
				 if(k1==0)	
				 {
				 	f1[i][j][k1][k2]=max(f1[i][j][k1][k2-1],f1[i][j+(1<<(k2-1))][k1][k2-1]);
				 	f2[i][j][k1][k2]=min(f2[i][j][k1][k2-1],f2[i][j+(1<<(k2-1))][k1][k2-1]);
				 }
				 else
				 {			 
                   f1[i][j][k1][k2]=max(f1[i][j][k1-1][k2],f1[i+(1<<(k1-1))][j][k1-1][k2]);
                   f2[i][j][k1][k2]=min(f2[i][j][k1-1][k2],f2[i+(1<<(k1-1))][j][k1-1][k2]);
//			    f1[i][j][k1][k2]=max(max(f1[i][j][k1-1][k2-1],f1[i+(1<<(k1-1))][j][k1-1][k2-1]),max(f1[i][j+(1<<(k2-1))][k1-1][k2-1],f1[i+(1<<(k1-1))][j+(1<<(k2-1))][k1-1][k2-1]));
//		     	f2[i][j][k1][k2]=min(min(f1[i][j][k1-1][k2-1],f1[i+(1<<(k1-1))][j][k1-1][k2-1]),min(f1[i][j+(1<<(k2-1))][k1-1][k2-1],f1[i+(1<<(k1-1))][j+(1<<(k2-1))][k1-1][k2-1]));
//		       
				}
			}
		}
	}
}

	
}
int find(int x,int y,int xx,int yy)
{
	int l1=log(xx-x+1)/log(2);
	int l2=log(yy-y+1)/log(2);
    int m1,m2,m3,m4;
    m1=f1[x][y][l1][l2];
    m2=f1[xx-(1<<l1)+1][y][l1][l2];
    m3=f1[x][yy-(1<<l2)+1][l1][l2];
    m4=f1[xx-(1<<l1)+1][yy-(1<<l2)+1][l1][l2];
    int maxn=max(max(m1,m2),max(m3,m4));
    
    m1=f2[x][y][l1][l2];
    m2=f2[xx-(1<<l1)+1][y][l1][l2];
    m3=f2[x][yy-(1<<l2)+1][l1][l2];
    m4=f2[xx-(1<<l1)+1][yy-(1<<l2)+1][l1][l2];
    int minn=min(min(m1,m2),min(m3,m4));
    
    return maxn-minn;
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n>>b>>k;
	for(int i=1;i<=n;i++)
	{
		for(int j=1;j<=n;j++)
		  cin>>a[i][j];
	}
	st();
//	for(int i=0;i<=3;i++)
//	{
//		print(i);
//		cout<<"----"<<endl;
//	}
//	cout<<endl<<" ----"<<endl;
	int x,y;

	while(k--)
	{
		cin>>x>>y;
		cout<<find(x,y,x+b-1,y+b-1)<<endl;	
	}
    return 0;
 } 

mood: Edge runners 请添加图片描述
an appointment on the moon

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

dai _ tu

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值