loj #10134. 「一本通 4.4 练习 1」Dis

Posted 2020-11-27 pjxpjx

题目链接

题意：

给定每个边的权值，每次求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;
}