HDU 2874 LCA RMQ + 변환 및 집합 검색

클릭 하여 링크 열기
제목: 두 도시 가 연결 되 어 있 는 지, 연결 되 지 않 은 출력 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
");continue;
            }
            int ans=dis[a]+dis[b]-2*dis[RMQ(a,b)];
            printf("%d
",ans);
        }
    }
    return 0;
}