程序師世界是廣大編程愛好者互助、分享、學習的平台,程序師世界有你更精彩!
首頁
編程語言
C語言|JAVA編程
Python編程
網頁編程
ASP編程|PHP編程
JSP編程
數據庫知識
MYSQL數據庫|SqlServer數據庫
Oracle數據庫|DB2數據庫
 程式師世界 >> 編程語言 >> C語言 >> C++ >> C++入門知識 >> HDU OJ 2586 How far away ?[LCA的Tarjan離線算法]

HDU OJ 2586 How far away ?[LCA的Tarjan離線算法]

編輯:C++入門知識

題意:給n個點,n-1條邊,保證任意兩點有且只有一條路,給m次查詢——每次查詢給兩個點  i ,j 求這兩點之間的距離。
思路:若用 一般的 最短路 來寫,時間復雜度比較高,會超時。這題時利用 LCA的Tarjan離線算法
LCA 就是 求 點  i,j 的最近公共祖先 k: d [ ] 代表根節點(可任取一點做根節點)到任意一點的距離,d [ i ]  +  d [ j ]  -    2*d [ k ]   、就是  i ,j  間的距離。
LCA的實現:   DFS+並查集      (假設求  i  j 的最近公共祖先)即為
                   在 DFS搜到 一個點(比如 i)時,判斷 是否 要求 該點與其他點(比如 j)的LCA。
                (1)若有,判斷 j 點是否 已經搜到過了?,若搜到過 並查集中 j 的 根(父)節點即為 i  和  j 的最近 公共祖 先!! 若 j 之前未 搜到,繼續向下搜索別的點(比如k),然後令 father[ k ]=i  ;
                 (2)沒有,繼續向下搜別的點(比如k),然後令 father[ k ]=i  ;
具體一些細節,參考代碼;
AC代碼:
[cpp] 
#pragma comment(linker, "/STACK:102400000,102400000") 
#include<stdio.h> 
#include<string.h> 
#include<vector> 
#include<queue> 
using namespace std; 
int d[410000],loop[410000],n,m; 
int father[410000],AC[400000]; 
const int inf=0x7fffffff; 
struct sb /*詢問*/ 

    int x; 
    int a; 
}; 
struct hello /*存地圖*/ 

    int x; 
    int d; 
}; 
struct node   /*隊列優先級*/ 

    friend bool operator<(node t1,node t2) 
    { 
        return t1.d>t2.d; 
    } 
    int x; 
    int d; 
}; 
vector<hello> V[41000]; 
vector<sb>F[41000]; 
void dis() 

    int a,b,i,j; 
    d[1]=0; 
    priority_queue<node> P; 
    node t1={1,0}; 
    P.push(t1); 
    while(!P.empty()) 
    { 
        i=P.top().x; 
        P.pop(); 
        if(loop[i]) 
            continue; 
        loop[i]=1; 
        for(a=0;a<V[i].size();a++) 
        { 
            int k=V[i][a].x; 
            if(loop[k]==0&&d[k]>d[i]+V[i][a].d) 
            { 
                d[k]=d[i]+V[i][a].d; 
                node t2={k,d[k]}; 
                P.push(t2); 
            } 
        } 
 
    } 
     

int find(int k) 

    if(k!=father[k]) 
        return father[k]=find(father[k]); 
    return father[k]; 

void init() 

    int a,b; 
    for(a=1;a<=n;a++) 
    { 
        father[a]=a; 
        d[a]=inf; 
        loop[a]=0; 
    } 

void dfs_lca(int k) 

    loop[k]=1; 
    int i,j; 
    for(i=0;i<F[k].size();i++) 
    { 
        if(loop[F[k][i].x]) 
        { 
            int tt=find(F[k][i].x); 
            //printf("-------%d %d   %d\n",k,F[k][i].x,tt); 
            AC[F[k][i].a]=d[k]+d[F[k][i].x]-2*d[tt]; 
        } 
    } 
    for(i=0;i<V[k].size();i++) 
    { 
        if(loop[V[k][i].x]==0) 
        { 
            dfs_lca(V[k][i].x); 
            father[V[k][i].x]=k; 
        } 
    } 

void output() 

    int i,j; 
    for(i=1;i<=m;i++) 
        printf("%d\n",AC[i]); 
    for(i=0;i<=n;i++) 
    { 
        V[i].clear(); 
        F[i].clear(); 
    } 

int main() 

    int a,b,c,ncase; 
    int x,y,z; 
    scanf("%d",&ncase); 
    for(c=1;c<=ncase;c++) 
    { 
        if(c>1) 
            printf("\n"); 
        scanf("%d%d",&n,&m); 
        init(); 
        for(a=1;a<n;a++) 
        { 
            scanf("%d%d%d",&x,&y,&z); 
            hello t1={y,z}; 
            V[x].push_back(t1); 
            hello t2={x,z}; 
            V[y].push_back(t2); 
        } 
        for(a=1;a<=m;a++) 
        { 
            scanf("%d%d",&x,&y); 
            if(x==y) 
            { 
                AC[a]=0; 
                continue; 
            } 
            sb t1={y,a}; 
            F[x].push_back(t1); 
            sb t2={x,a}; 
            F[y].push_back(t2); 
        } 
        dis(); 
        memset(loop,0,sizeof(loop)); 
        dfs_lca(1); 
        output(); 
    } 

/*
12  12
1 2 3
1 3 3
2 4 3
2 5 3
2 6 3
3 7 3
3 8 3
4 9 3
5 10 3
5 11 3
7 12 3
 
4 1
4 2
4 3
4 4
5 6
9 8
11 2
8 2
7 8
9 3
10 11
5 10
*/ 
作者:PIAOYI0208

  1. 上一頁:
  2. 下一頁:
Copyright © 程式師世界 All Rights Reserved