hdu2586 LCA模板

来源：互联网发布：jsp import java类编辑：程序博客网时间：2024/06/10 02:11

hdu2586

先预处理

然后是在线询问（也就是一边读入一边查询）

#pragma comment(linker, "/STACK:1024000000,1024000000")#include <cstdio>#include <iostream>#include <vector>using namespace std;#define maxn 400010int n;struct Node{    int v, w;    Node(int v_, int w_){        v= v_, w= w_;    }};vector<Node>g[maxn];int father[maxn];int dis[maxn];int dep[maxn];void dfs(int u, int fa, int deep){    father[u]= fa;    dep[u]= deep;    for(int i=0; i<g[u].size(); ++i){        int v= g[u][i].v;        if(v==fa) continue; //        dis[v]= dis[u] + g[u][i].w;        dfs(v, u, deep+1);    }}int p[maxn][30];void Init_LCA(){    for(int j=0; (1<<j)<=n; ++j)        for(int i=1; i<=n; ++i)            p[i][j]= -1;    for(int i=1; i<=n; ++i) p[i][0]= father[i];    for(int j=1; (1<<j)<=n; ++j)        for(int i=1; i<=n; ++i)            if( p[i][j-1] != -1 )                p[i][j]= p[ p[i][j-1] ][ j-1 ];}int LCA(int x, int y){    if( dep[x] < dep[y] ) swap(x, y);    int i,lg;    for(lg=0; (1<<lg)<=dep[x]; ++lg);    --lg;    /// 使x往上走直到和y在同一水平线上；    for(i=lg; i>=0; --i)        if( dep[x] - (1<<i) >= dep[y] )            x= p[x][i];    if( x==y ) return x;    /// 此时x，y在同一水平线上，使x，y同时以相同的速度(2^j)往上走；    for(i=lg; i>=0; --i)        if( p[x][i]!=-1 && p[x][i]!=p[y][i] )            x= p[x][i], y= p[y][i];    return father[x];}int main(){    int T, m;    cin>>T;    while(T--){        cin>>n>>m;        for(int i=1; i<=n; ++i) g[i].clear();        for(int i=1; i<n; ++i){            int x, y, w;            scanf("%d%d%d", &x,&y,&w);            g[x].push_back( Node(y, w) );            g[y].push_back( Node(x, w) );        }        dis[1]= 0;        dfs(1, -1, 0);        Init_LCA();        while(m--){            int x, y;            scanf("%d%d", &x, &y);            printf("%d\n", dis[x] + dis[y] - 2*dis[ LCA(x, y) ]);        }    }}

LCA离线算法(储存所有询问之后再查询)

http://blog.csdn.net/dellaserss/article/details/8250926