D13 点的距离 LCA应用

视频链接:

// #10130. 「一本通 4.4 例 1」点的距离 https://lov.ac/p/10130
///////////////////倍增算法  
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N=100010,h=18;
int n,m,a,b;
vector<int> e[N];
int dep[N],fa[N][19],dis[N];

void dfs(int u,int father){
  dep[u]=dep[father]+1;
  fa[u][0]=father;
  for(int i=1; i<=h; i++)
    fa[u][i]=fa[fa[u][i-1]][i-1];
  for(int v:e[u]){
    if(v==father) continue;
    dis[v]=dis[u]+1;        
    dfs(v,u);
  }
}
int lca(int x,int y){
  if(dep[x]<dep[y]) swap(x,y);
  for(int i=h; i>=0; i--)
    if(dep[fa[x][i]]>=dep[y])
      x=fa[x][i];
  if(x==y) return x;
  for(int i=h; i>=0; i--)
    if(fa[x][i]!=fa[y][i]) 
      x=fa[x][i],y=fa[y][i];
  return fa[x][0];
}
int main(){
  scanf("%d",&n);
  for(int i=1; i<n; i++){
    scanf("%d%d",&a,&b);
    e[a].push_back(b);
    e[b].push_back(a);
  }
  dfs(1,0);
  scanf("%d",&m);
  while(m--){
    scanf("%d%d",&a,&b);
    int d=dis[a]+dis[b]-dis[lca(a,b)]*2;
    printf("%d\n",d);
  }
  return 0;
}

///////////////////树链剖分  
#include <iostream>
#include <cstring>
#include <algorithm>
#include <vector>
using namespace std;
const int N=100010;
int n,m,a,b,c;
vector<int> e[N];
int dep[N],fa[N],son[N],sz[N],dis[N];
int top[N];

void dfs1(int u,int father){
  fa[u]=father,dep[u]=dep[father]+1,sz[u]=1;
  for(int v:e[u]){
    if(v==father) continue;
    dis[v]=dis[u]+1;
    dfs1(v,u);
    sz[u]+=sz[v];
    if(sz[son[u]]<sz[v])son[u]=v;
  }
}
void dfs2(int u,int t){
  top[u]=t;
  if(!son[u]) return;
  dfs2(son[u],t);
  for(int v:e[u]){
    if(v==fa[u]||v==son[u])continue;
    dfs2(v,v);
  }
}
int lca(int x,int y){
  while(top[x]!=top[y]){
    if(dep[top[x]]<dep[top[y]])swap(x,y);
    x=fa[top[x]];
  }
  return dep[x]<dep[y]?x:y;
}
int main(){
  scanf("%d",&n);
  for(int i=1; i<n; i++){
    scanf("%d%d",&a,&b);
    e[a].push_back(b);
    e[b].push_back(a);
  }
  dfs1(1,0);
  dfs2(1,1);
  scanf("%d",&m);
  while(m--){
    scanf("%d%d",&a,&b);
    int d=dis[a]+dis[b]-dis[lca(a,b)]*2;
    printf("%d\n",d);
  }
  return 0;
}
///////////////////Tarjan 算法 
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
typedef pair<int,int> PII;
const int N=100010,M=N*2;
int n,m,a,b;
vector<int> e[N];
vector<PII> query[N];
int fa[N],vis[N],dis[N];
int ans[M];

int find(int u){
  if(fa[u]==u) return u;
  return fa[u]=find(fa[u]);
}
void tarjan(int u){
  vis[u]=1;
  for(int v:e[u]){
    if(vis[v]) continue;
    dis[v]=dis[u]+1;
    tarjan(v);
    fa[v]=u;
  }
  for(auto ed:query[u]){
    int v=ed.first,i=ed.second;
    if(vis[v])
      ans[i]=dis[u]+dis[v]-dis[find(v)]*2;
  }    
}
int main(){
  scanf("%d",&n);
  for(int i=1; i<n; i++){
    scanf("%d%d",&a,&b);
    e[a].push_back(b);
    e[b].push_back(a);
  }
  scanf("%d",&m);
  for(int i=1; i<=m; i++){
    scanf("%d%d",&a,&b);
    query[a].push_back({b,i});
    query[b].push_back({a,i});
  }
  for(int i=1;i<=n;i++) fa[i]=i;
  tarjan(1);
  for(int i=1; i<=m; i++)
    printf("%d\n",ans[i]);
  return 0;
}

 

 

posted @ 2022-05-28 13:27  董晓  阅读(539)  评论(0编辑  收藏  举报