st表+lca模板

st表

#include<bits/stdc++.h>
#define ll long long
#define fd(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
ll n,m,t,l,r,dp[1000100][100];
int main()
{
	std::ios::sync_with_stdio(false);
	cin>>m>>n;
	fd(i,1,m) cin>>dp[i][0];
	t=log(m)/log(2)+1;
	fd(j,1,t)
	{
		fd(i,1,m-(1<<j)+1)
		{
			dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
		}
	}
	fd(i,1,n)
	{
		cin>>l>>r;
		ll tt=log(r-l+1)/log(2);
		printf("%lld\n",max(dp[l][tt],dp[r-(1<<tt)+1][tt]));
	}
	return 0;
}

LCA

#include<bits/stdc++.h>
#define ll long long
#define fd(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
ll n,m,s,tot,head[1000100],aa,bb;
ll d[1000100],f[1000100][20],t;
struct edge
{
	ll to,nxt,w;
}e[1000100];
void add(ll a,ll b,ll c)
{
	e[++tot].nxt=head[a];
	e[tot].to=b;
	e[tot].w=c;
	head[a]=tot;
}
void bfs()
{
	queue<ll> q;
	q.push(s);
	d[s]=1;
	while(!q.empty())
	{
		ll x=q.front();
		q.pop();
		for(ll i=head[x];i;i=e[i].nxt)
		{
			ll y=e[i].to;
			if(d[y]) continue;
			d[y]=d[x]+1;
			f[y][0]=x;
			fd(j,1,t)
			{
				f[y][j]=f[f[y][j-1]][j-1];
			}
			q.push(y);
		}
	}
}
ll lca(ll a,ll b)
{
	if(d[a]>d[b]) swap(a,b);
	for(ll i=t;i>=0;--i)
	{
		if(d[f[b][i]]>=d[a]) b=f[b][i];
	}
	if(a==b) return a;
	for(ll i=t;i>=0;--i)
	{
		if(f[a][i]!=f[b][i]) a=f[a][i],b=f[b][i];
	}
	return f[a][0];
}
int main()
{
//	freopen("pow.in","r",stdin);
//	freopen("pow.out","w",stdout);
	cin>>n>>m>>s;
	fd(i,1,n-1)
	{
		cin>>aa>>bb;
		add(aa,bb,0);
		add(bb,aa,0);
	}
	t=log(n)/log(2);
	bfs();
	fd(i,1,m)
	{
		cin>>aa>>bb;
		cout<<lca(aa,bb)<<endl;
	}
	return 0 ;
}
posted @ 2024-03-01 19:45  whrwlx  阅读(9)  评论(0编辑  收藏  举报