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