标签：dist 1171 idx int void ne 距离 st lca

 题意:求树上两点的最短距离。考查lca，预处理点到根的距离dist，则两点a、b之间的最短距离是dist[a]+dist[b]-2*dist[p];p是a、b最近公共祖先。

 关于lcahttps://blog.csdn.net/my_sunshine26/article/details/72717112

#include<bits/stdc++.h>
using namespace std;
const int N = 10005;
const int M = 2*N;
int e[M],ne[M],w[M],h[M];
int dist[N];
int res[N];
int st[N];
int f[N];
int idx;
vector<pair<int,int >>q[M];
int find(int a){
    if(f[a]!=a)f[a]=find(f[a]);
    return f[a];
}
void add(int a,int b,int v){
    e[idx]=b;ne[idx]=h[a];w[idx]=v;h[a]=idx++;
}
void dfs(int u,int f){
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(j!=f){
            dist[j]=dist[u]+w[i];
            dfs(j,u);
        }
    }
}
void tarjan(int u){
    st[u]=1;
    for(int i=h[u];~i;i=ne[i]){
        int j=e[i];
        if(!st[j]){
            tarjan(j);
            f[j]=u;
        }
    }
    for(int i=0;i<q[u].size();i++){
        int to=q[u][i].first;
        int pos=q[u][i].second;
        int anc=find(u);
        if(st[to]==2){
            res[pos]=dist[u]+dist[to]-2*dist[anc]; 
        }
    }
    st[u]=2;
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)f[i]=i;
    memset(h,-1,sizeof(h));
    for(int i=1;i<n;i++){
        int a,b,v;
        scanf("%d%d%d",&a,&b,&v);
        add(a,b,v);
        add(b,a,v);
    }
    for(int i=0;i<m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        q[x].push_back(make_pair(y,i));
        q[y].push_back(make_pair(x,i));
    }
    dfs(1,-1);
    tarjan(1);
    for(int i=0;i<m;i++){
        printf("%d\n",res[i]);
    }
    return 0;
}

标签：dist,1171,idx,int,void,ne,距离,st,lca
来源： https://www.cnblogs.com/mohari/p/13946958.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。