【2586】How far away ？

spe1437

于 2021-08-20 22:16:09 发布

阅读量71

点赞数

分类专栏： CCPC比赛练习

本文链接：https://blog.csdn.net/weixin_49646622/article/details/119832398

版权

CCPC比赛练习专栏收录该内容

46 篇文章 0 订阅

订阅专栏

本文介绍了如何使用LCA（最近公共祖先）与DFS结合，并查集和树链剖分的方法解决图论问题。在C++代码中，展示了如何初始化、查找、合并并查集，并通过Tarjan算法进行深度优先搜索，计算节点间的最短路径。该算法适用于处理树状结构的数据，例如在树的边权更新和查询中最短路径的问题。

摘要由CSDN通过智能技术生成

2586
LCA（并查集）+DFS,树链剖分

#include<bits/stdc++.h>
using namespace std;
const int N = 50005;
vector<int> v[N],w[N],query[N],num[N];
int pre[N],dist[N],ans[N];
bool vis[N];
int n;

void Init(){
    for(int i=1; i<=n; i++){
        v[i].clear();
        w[i].clear();
        query[i].clear();
        num[i].clear();
        pre[i] = i;
        dist[i] = 0;
        vis[i] = false;
    }
}

int Find(int x){
    if(pre[x] != x)
        pre[x] = Find(pre[x]);
    return pre[x];
}

void Union(int x,int y){
    x = Find(x);
    y = Find(y);
    if(x == y) return;
    pre[y] = x;
}

void Tarjan(int cur,int val){
    vis[cur] = true;
    dist[cur] = val;
    int size = v[cur].size();
    for(int i=0;i<size;i++){
        int tmp = v[cur][i];
        if(vis[tmp]) continue;
        Tarjan(tmp,val + w[cur][i]);
        Union(cur,tmp);
    }
    int Size = query[cur].size();
    for(int i=0;i<Size;i++){
        int tmp = query[cur][i];
        if(!vis[tmp]) continue;
        ans[num[cur][i]] = dist[cur] + dist[tmp] - 2*dist[Find(tmp)];
    }
}

int main(){
    int T,Q,x,y,z;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&Q);
        Init();
        for(int i=1;i<n;i++){
            scanf("%d%d%d",&x,&y,&z);
            v[x].push_back(y);
            w[x].push_back(z);
            v[y].push_back(x);
            w[y].push_back(z);
        }
        for(int i=0;i<Q;i++){
            scanf("%d%d",&x,&y);
            query[x].push_back(y);
            query[y].push_back(x);
            num[x].push_back(i);
            num[y].push_back(i);
        }
        Tarjan(1,0);
        for(int i=0;i<Q;i++)
           printf("%d\n",ans[i]);
    }
    return 0;
}