loj #10134. 「一本通 4.4 练习 1」Dis
作者:互联网
题目链接
题意:
给定每个边的权值,每次求u,v两点的距离。
题解:
把每个边上的权值给挂到每条边的儿子节点下面,跑一边树上前缀和自顶向下模板,没了。
裸的树上前缀和模板
但要注意:DFS中邻接表中前缀和的变量不能打错了
Code:
#include <iostream>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <queue>
using namespace std;
const int N=200005;
const int M=25;
int Q,n,m,head[N],dou[N][M],cnt,dep[N],b[N],k,total[N],wei[N];
struct node{
int u,v,next,sum;
}ed[N];
void add_edge(int u,int v,int k)
{
cnt++;
ed[cnt].u=u;
ed[cnt].v=v;
ed[cnt].next=head[u];
ed[cnt].sum=k;
head[u]=cnt;
}
void dfs(int xx)
{
for(int i=1;i<=20;i++)
{
dou[xx][i]=dou[dou[xx][i-1]][i-1];
}
for(int i=head[xx];i!=0;i=ed[i].next)
{
int temp=ed[i].v;
if(b[temp]==0)
{
b[temp]=1;
dep[temp]=dep[xx]+1;
total[temp]=ed[i].sum+total[xx];
dou[temp][0]=xx;
dfs(temp);
}
}
}
int LCA(int x,int y)
{
if(dep[x]<dep[y])
swap(x,y);
int temp=dep[x]-dep[y];
for(int i=20;i>=0;i--)
{
if(1<<i&temp)
{
x=dou[x][i];
}
}
if(x==y)
return x;
for(int i=20;i>=0;i--)
{
if(dou[x][i]!=dou[y][i])
{
x=dou[x][i];
y=dou[y][i];
}
}
return dou[x][0];
}
int main()
{
cin>>n;
cin>>m;
for(int i=1;i<n;i++)
{
int x,y,k;
scanf("%d%d%d",&x,&y,&k);
add_edge(x,y,k);
add_edge(y,x,k);
}
b[1]=1;
dep[1]=1;
dfs(1);
while(m--)
{
int x,y;
scanf("%d%d",&x,&y);
int lca=LCA(x,y);
cout<<total[x]+total[y]-total[lca]-total[lca]<<endl;
}
return 0;
}
标签:10134,4.4,cnt,int,ed,head,loj,include,dou 来源: https://www.cnblogs.com/pjxpjx/p/12401391.html