本文主要是介绍HDU - 2586 How far away LCA+tanjar离线算法,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
传送门
题意:求树上的两点的距离,放在一棵树上,而且题目提示只有一条,很明显是这条路线一定会经过他们的最近公共祖先啦,然后就是一个求距离的问题了。如果要采用离线算法的话,就是通过处理所有询问,那么肯定不能单独的求两点之间的距离,我们肯定是要求每一个点到root节点的距离。利用tanjan算法跑一边就可以。
设root为根结点,询问a,b两点距离,a,b的公共祖先为lca点。dis数组为每个点到根节点的距离。
那么ans=dis[a]+dis[b]-2*dis[lca];
具体做法如下
#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 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))
typedef long long ll;
using namespace std;
const int maxn=4e5+5;
struct node
{int to,p,val,lca;
}edge[maxn],qedge[maxn];
int vis[maxn],d[maxn],head[maxn],qhead[maxn],f[maxn],s[maxn],e[maxn];
int sign,qsign,n,m;
int find(int x)
{if(x==f[x]) return x;else return f[x]=find(f[x]);
}
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 init()
{sign=qsign=0;for(int i=1;i<=n;i++){f[i]=i;head[i]=qhead[i]=-1;vis[i]=d[i]=0;}
}
void dfs(int u)
{vis[u]=1;for(int i=head[u];i!=-1;i=edge[i].p){int v=edge[i].to;if(!vis[v]){d[v]=d[u]+edge[i].val;dfs(v);f[v]=u;}}for(int i=qhead[u];i!=-1;i=qedge[i].p){int v=qedge[i].to;if(vis[v]){qedge[i].lca=find(v);if(i%2==0)qedge[i+1].lca=qedge[i].lca;elseqedge[i-1].lca=qedge[i].lca;}}
}
int main()
{int t,x,y,z;scanf("%d",&t);while(t--){scanf("%d %d",&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",&x,&y);s[i]=x;e[i]=y;qadd(x,y);qadd(y,x);}dfs(1);for(int i=0;i<m;i++){printf("%d\n",d[s[i+1]]+d[e[i+1]]-2*d[qedge[2*i].lca]);}}return 0;
}
这篇关于HDU - 2586 How far away LCA+tanjar离线算法的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!