HDU 2586 How far away ? LCA ---tanjar+并查集 离线算法
生活随笔
收集整理的這篇文章主要介紹了
HDU 2586 How far away ? LCA ---tanjar+并查集 离线算法
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
tanjar算法離線求LCA的思想主要是利用并查集的思想。
求距離的話就是d[start[i]]+end[en[i]]-2*d[lca[i]];
首先從根節點dfs,在深度遍歷的回溯的過程中不斷的更新自己的父節點,使得查詢兩個點肯定是存在一顆最近的子樹里邊,所以
find(v)就是LCA。
void dfs(int u)
{f[u]=u;//首先把該點的父親賦值為自己,即記錄父親又當vis數組for(int i=head[u];~i;i=edge[i].p){int v=edge[i].to;if(!f[v]) ///f[]表示父親,一開始賦為0,只要遍歷過就肯定不為0 (init里)d[v]=d[u]+edge[i].val;//記錄到根節點的距離dfs(v);///如果沒遍歷過就繼續跑f[v]=u;///重點,思想就是上邊一步遍歷v時f[v]還是自己,這個遍歷完了之后才把父親u賦值給v,/// 當下邊存在x一個詢問到v,v和x一直都會是在一棵樹下,并且find(v)的值就是最v,x的///最近的父節點,也就是lca}for(int i=qhead[u];~i;i=qedge[i].p)//詢問是否有邊一樣{int v=qedge[i].to;if(f[v]) lca[i/2+1]=find(v);///因為加入邊的時候加入了兩次,并且從0開始加,所以是i/2+1。}
}
AC代碼,是要比倍增快一點。
#include<stdio.h>
#include<string.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<set>
#include<stack>
#include<vector>
#include<map>
#include<queue>
#define myself i,l,r
#define lson i<<1
#define rson i<<1|1
#define Lson i<<1,l,mid
#define Rson i<<1|1,mid+1,r
#define half (l+r)/2
#define inff 0x3f3f3f3f
#define lowbit(x) x&(-x)
#define PI 3.14159265358979323846
#define me(a,b) memset(a,b,sizeof(a))
#define min4(a,b,c,d) min(min(a,b),min(c,d))
#define min3(x,y,z) min(min(x,y),min(y,z))
const int dir[4][2]= {0,-1,-1,0,0,1,1,0};
typedef long long ll;
const ll inFF=9223372036854775807;
typedef unsigned long long ull;
using namespace std;
const int maxn=4e5+6;
int head[maxn],qhead[maxn],f[maxn],lca[maxn],d[maxn],st[maxn],en[maxn];
int sign,qsign,n,m;
struct node
{int to,p,val;
}edge[maxn],qedge[maxn];
int find(int x)
{return x==f[x]?x:f[x]=find(f[x]);
}
void init()
{sign=qsign=d[1]=0;for(int i=0;i<=2*n;i++){head[i]=qhead[i]=-1;f[i]=0;}
}
void add(int u,int v,int val)
{edge[sign]=node{v,head[u],val};head[u]=sign++;
}
void qadd(int u,int v)
{qedge[qsign]=node{v,qhead[u]};qhead[u]=qsign++;
}
void dfs(int u)
{f[u]=u;for(int i=head[u];~i;i=edge[i].p){int v=edge[i].to;if(!f[v]) d[v]=d[u]+edge[i].val,dfs(v),f[v]=u;}for(int i=qhead[u];~i;i=qedge[i].p){int v=qedge[i].to;if(f[v]) lca[i/2+1]=find(v);}
}
int main()
{int t,x,y,z;cin>>t;while(t--){cin>>n>>m;init();for(int i=1;i<n;i++) scanf("%d %d %d",&x,&y,&z),add(x,y,z),add(y,x,z);for(int i=1;i<=m;i++) scanf("%d %d",&st[i],&en[i]),qadd(st[i],en[i]),qadd(en[i],st[i]);dfs(1);for(int i=1;i<=m;i++) printf("%d\n",d[st[i]]+d[en[i]]-2*d[lca[i]]);}return 0;
}
?
總結
以上是生活随笔為你收集整理的HDU 2586 How far away ? LCA ---tanjar+并查集 离线算法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: POJ - 3160 Father Ch
- 下一篇: POJ - 1986 Distance