HDU 2874 LCA转RMQ+并查集

来源：互联网发布：windows命令ftp放文件编辑：程序博客网时间：2024/06/06 14:26

点击打开链接

题意：问两个城市是否相连，不相连输出Not connected，否则输出两个城市间的最短距离

思路：用并查集判断两个城市的连通性，如果联通则做法和LCA一样，但是注意的一点是地图不连通的话，我们要将所有点都建起来，就要加一个模拟的点，将所有图串起来，很好处理的，看一下就会了

#include <vector>#include <stdio.h>#include <string.h>#include <stdlib.h>#include <iostream>#include <algorithm>using namespace std;typedef long long ll;const int inf=0x3f3f3f3f;const int maxn=10010;struct edge{    int to,cost;    edge(int a,int b){to=a;cost=b;}};vector<edge>G[maxn];int dp[maxn*2][20],L[maxn*2],E[maxn*2],dis[maxn],H[maxn],f[maxn];bool vis[maxn];int n,k;void dfs(int t,int deep){    k++;E[k]=t;L[k]=deep;H[t]=k;    for(unsigned int i=0;i<G[t].size();i++){        edge e=G[t][i];        if(!vis[e.to]){            dis[e.to]=dis[t]+e.cost;            vis[e.to]=1;            dfs(e.to,deep+1);            k++;E[k]=t;L[k]=deep;        }    }}void RMQ_init(){    for(int i=1;i<=2*(n+1)-1;i++) dp[i][0]=i;    for(int i=1;(1<<i)<=2*(n+1)-1;i++){        for(int j=1;j+(1<<i)-1<=2*n-1;j++){            if(L[dp[j][i-1]]<L[dp[j+(1<<(i-1))][i-1]]) dp[j][i]=dp[j][i-1];            else dp[j][i]=dp[j+(1<<(i-1))][i-1];        }    }}int RMQ(int le,int ri){    le=H[le];ri=H[ri];    if(le>ri) swap(le,ri);    int kk=0;    while((1<<(kk+1))<=ri-le+1) kk++;    if(L[dp[le][kk]]<L[dp[ri-(1<<kk)+1][kk]]) return E[dp[le][kk]];    else return E[dp[ri-(1<<kk)+1][kk]];}int find1(int a){    if(a!=f[a])        f[a]=find1(f[a]);    return f[a];}void unite(int a,int b){    int aa=find1(a);    int bb=find1(b);    f[aa]=bb;}int main(){    int T,m,a,b,c;    while(scanf("%d%d%d",&n,&m,&T)!=-1){        for(int i=0;i<=n;i++) f[i]=i;        memset(dis,0,sizeof(dis));        memset(vis,0,sizeof(vis));        for(int i=0;i<maxn;i++) G[i].clear();        for(int i=0;i<m;i++){            scanf("%d%d%d",&a,&b,&c);            unite(a,b);            G[a].push_back(edge(b,c));            G[b].push_back(edge(a,c));        }        for(int i=1;i<=n;i++){            if(f[i]==i){                G[0].push_back(edge(i,0));                G[i].push_back(edge(0,0));            }        }        k=0;vis[0]=1;        dfs(0,1);RMQ_init();        while(T--){            scanf("%d%d",&a,&b);            if(find1(a)!=find1(b)){                printf("Not connected\n");continue;            }            int ans=dis[a]+dis[b]-2*dis[RMQ(a,b)];            printf("%d\n",ans);        }    }    return 0;}

0 0