lca的裸题 求无向图上两点的最短距离 查询两点的lca然后就是两点分别到达根节点的距离减去两倍lca到根节点的距离,lca随意搞,纯暴力的在线查询也能过 然后是二分查找或者基于rmq的lca查询或者tarjan
先是暴力
//#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define maxn 40010
int n;
struct node{
int v,w;
node(int v,int w):v(v),w(w){}
};
vector<node>g[maxn];
int fa[maxn],dis[maxn],dep[maxn];
void dfs(int u,int f,int deep)
{
fa[u]=f;
dep[u]=deep;
for(int i=0;i<g[u].size();++i)
{
int v=g[u][i].v;
if(f==v)continue;
dis[v]=dis[u]+g[u][i].w;
dfs(v,u,deep+1);
}
}
int lca(int u,int v)
{
while(dep[u]>dep[v]){u=fa[u];}
while(dep[v]>dep[u]){v=fa[v];}
while(u!=v)
{
u=fa[u];
v=fa[v];
}
return u;
}
int main()
{
int t,m;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i)g[i].clear();
for(int i=1;i<n;++i){
int x,y,w;
scanf("%d%d%d",&x,&y,&w);
g[x].push_back(node(y,w));
g[y].push_back(node(x,w));
}
dis[1]=0;
dfs(1,-1,0);
while(m--)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
}
}
return 0;
}
然后是二分查找:
#include<cstdio>
#include<iostream>
#include<vector>
using namespace std;
#define maxn 400010
int n;
struct node{
int v,w;
node(int v,int w):v(v),w(w){}
};
vector<node>g[maxn];
int fa[maxn],dis[maxn],dep[maxn];
void dfs(int u,int f,int deep)
{
fa[u]=f;
dep[u]=deep;
for(int i=0;i<g[u].size();++i)
{
int v=g[u][i].v;
if(f==v)continue;
dis[v]=dis[u]+g[u][i].w;
dfs(v,u,deep+1);
}
}
int p[maxn][30];
void init()
{
for(int j=0;(1<<j)<=n;++j)
for(int i=1;i<=n;++i)
p[i][j]=-1;
for(int i=1;i<=n;++i)p[i][0]=fa[i];
for(int j=1;(1<<j)<=n;++j)
for(int i=1;i<=n;++i)
if(p[i][j]!=-1)
p[i][j]=p[p[i][j-1]][j-1];
}
int lca(int x,int y)
{
if(dep[x]<dep[y])swap(x,y);
int lg;
for(lg=0;(1<<lg)<=dep[x];++lg);
--lg;
for(int i=lg;i>=0;i--)
if(dep[x]-(1<<i)>=dep[y])
x=p[x][i];
if(x==y)return x;
for(int i=lg;i>=0;--i)
if(p[x][i]!=-1&&p[x][i]!=p[y][i])
x=p[x][i],y=p[y][i];
return fa[x];
}
int main()
{
int t,m;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i)g[i].clear();
for(int i=1;i<n;++i){
int x,y,w;
scanf("%d%d%d",&x,&y,&w);
g[x].push_back(node(y,w));
g[y].push_back(node(x,w));
}
dis[1]=0;
dfs(1,-1,0);
init();
while(m--)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
}
}
return 0;
}