这两天学会了树上倍增求lca,感觉树上倍增在线求法好方便呀,而且很快,利用二进制拆分的思路和倍增思想,感觉挺不错,然后验了几道题感觉还不错,我写的板子解决一些题应该没什么大问题,学到新东西,心里怪开心的
AC代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
//lca板子题,求俩个点最短距离
//树上两点最短路径:从根节点出发dis[u]+dis[v]-dis[lca]*2
const int maxn=1e5+5;
struct node
{
int to,nex,w;
} road[maxn*2];
int n,q,cnt;
int pre[maxn][32],head[maxn],depth[maxn];
int dis[maxn];
void add(int u,int v,int w)
{
road[cnt].to=v;
road[cnt].w=w;
road[cnt].nex=head[u];
head[u]=cnt++;
}
void dfs(int u,int fa)
{
pre[u][0]=fa;
depth[u]=depth[fa]+1;
for(int i=1; (1<<i)<=depth[u]; i++) //倍增
pre[u][i]=pre[pre[u][i-1]][i-1];
for(int i=head[u]; ~i; i=road[i].nex)
{
int v=road[i].to;
if(v!=fa)
{
dis[v]=dis[u]+road[i].w;
dfs(v,u);
}
}
}
int lca(int u,int v)
{
if(depth[u]<depth[v])
{
swap(u,v);
}
int i=-1,j;
while((1<<(i+1))<=depth[u])
i++;
for(j=i; j>=0; j--)
{
if(depth[u]-(1<<j)>=depth[v])
{
u=pre[u][j];
}
}
if(u==v)
return u;
for(int j=i; j>=0; j--)
{
if(pre[u][j]!=pre[v][j])
{
u=pre[u][j];
v=pre[v][j];
}
}
return pre[u][0];
}
int main()
{
int m;
scanf("%d %d",&n,&m);
memset(head,-1,sizeof(head));
memset(depth,0,sizeof(depth));
cnt=0;
for(int i=1; i<=m; i++)
{
int u,v,w;
char s[20];
scanf("%d %d %d %s",&u,&v,&w,s);
add(u,v,w);
add(v,u,w);
}
dis[1]=0;
dfs(1,0);
scanf("%d",&q);
while(q--)
{
int u,v;
scanf("%d %d",&u,&v);
int lc=lca(u,v);
int ans=dis[u]+dis[v]-2*dis[lc];
printf("%d\n",ans);
}
}