LCA 最近公共祖先 模板
阿新 • • 發佈:2018-11-30
Lca 離線演算法(tarjan演算法) 模板
#include<bits/stdc++.h>
#define _ 1000100
using namespace std;
char buf[1<<15],*fs,*ft;
inline char getc(){return (ft==fs&&(ft=(fs=buf)+fread(buf,1,1<<15,stdin),ft==fs))?0:*fs++;}
inline int read()
{
int x=0,f=1; char ch=getc();
while(!isdigit(ch) ) {if(ch=='-') f=-1; ch=getc();}
while(isdigit(ch)) {x=(x<<1)+(x<<3)+(ch^48); ch=getc();}
return x*f;
}
void put(int x)
{
if(x==0){putchar('0');putchar('\n');return;}
if(x<0){putchar('-');x=-x;}
int num=0;char ch[16];
while(x) ch[++num]=x%10+'0',x/=10;
while(num) putchar(ch[num--] );
putchar('\n');
}
int ver1[_],Next1[_],link1[_];
int ver2[_],Next2[_],link2[_],id[_];
int tot=0,n,m,ans[_],fa[_],vis[_];
inline void add1(int x,int y)
{
ver1[++tot]=y,Next1[tot]=link1[x],link1[x]=tot;
}
inline void add2(int x,int y,int t)
{
ver2[++tot]=y,Next2[tot]=link2[x],link2[x]=tot,id[tot]= t;
}
int get(int x)
{
if(fa[x]==x)return x;
return fa[x]=get(fa[x]);
}
void Lca(int x)
{
vis[x]=1;
for(int i=link1[x];i;i=Next1[i])
{
int y=ver1[i];
if(vis[y]) continue;
Lca(y);
fa[y]=x;
}
for(int i=link2[x];i;i=Next2[i])
{
int y=ver2[i];
if(vis[y])
ans[id[i]]=get(y);
}
}
int main()
{
n=read();m=read();
for(int i=1;i<n;++i)
{
int x=read(),y=read();
add1(x,y),add1(y,x);
}
tot=0;
for(int i=1;i<=m;++i)
{
int x=read(),y=read();
add2(x,y,i),add2(y,x,i);
}
for(int i=1;i<=n;++i)
fa[i]=i;
Lca(1);
for(int i=1;i<=m;++i)
put(ans[i]);
return 0;
}
Lca 樹上倍增 模板
抄演算法競賽進階指南上的模板
#include<bits/stdc++.h>
using namespace std;
inline int read()
{
int f=1,num=0;
char ch=getchar();
while (ch<'0'||ch>'9') { if (ch=='-') f=-1; ch=getchar(); }
while (ch>='0'&&ch<='9') num=(num<<1)+(num<<3)+ch-'0', ch=getchar();
return num*f;
}
const int SIZE=50010;
int f[SIZE][20],d[SIZE],dist[SIZE];
int ver[SIZE*2],Next[SIZE*2],edge[SIZE*2],head[SIZE];
int T,n,m,tot,t;
queue<int>q;
void add(int x,int y,int z)
{
ver[++tot]=y,edge[tot]=z,Next[tot]=head[x],head[x]=tot;
}
void bfs()
{//預處理
q.push(1);
d[1]=1;
while (!q.empty())
{
int x=q.front();
q.pop();
for (int i=head[x];i;i=Next[i])
{
int y=ver[i];
if (d[y]) continue;
d[y]=d[x]+1;
dist[y]=dist[x]+edge[i];
f[y][0]=y;
for (int j=1;j<=t;j++)
f[y][j]=f[f[y][j-1]][j-1];
q.push(y);
}
}
}
int lca(int x,int y)//回答一個詢問
{
if (d[x]<d[y]) swap(x,y);
for (int i=t;i>=0;i--)
if (d[f[y][i]]>=d[x]) y=f[y][i];
if (x==y) return x;
for (int i=t;i>=0;i--)
if (f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
int main()
{
T=read();
while (T--)
{
n=read(),m=read();
t=(int)(log(n)/log(2))+1;
//清空
for (int i=1;i<=n;++i)
head[i]=d[i]=0;
tot=0;
//讀入一棵樹
for (int i=1;i<n;++i)
{
int x=read(),y=read(),z=read();
add(x,y,z),add(y,x,z);
}
bfs();
//回答問題
for (int i=1;i<=m;++i)
{
int x=read(),y=read();
printf("%d\n",dist[x]+dist[y]-2*dist[lca(x,y)]);
}
}
return 0;
}