HDU2874【LCA(模板)】

来源:互联网 发布:java代码的移植性 编辑:程序博客网 时间:2024/06/06 00:47
第一题LCA,代码参考自:Ice_Crazy

思路:

这个最短路算法是想都别想了,可以看出这幅图就是树嘛,那么对于查询就是求树上两个结点最短距离。

这里就是利用LCA的tarjan离线算法。

算法的大致流程:

对于每一点u,

①  :建立以u为代表元素的集合。

②  :遍历与u相连的结点v,如果没有访问过,对与v使用Tarjan-LCA算法,结束后,将v的集合并入u的集合。

③  :对于与u相关的询问(u,v),如果v被访问过,则结果就是v所在集合的代表。

在这里还需要算距离,需要深度:dis<v1,v2>=dis[v1]+dis[v2]-2*dis[lca];

简要说下这份代码的几个变量的作用:

vis[ ]是用来确定集合,可能存在多棵树?

pre[ ]每次存的是前驱,为什么正好在Find()过程中就他们的LCA呢?

这是因为一直在处理的是子树呀!对于结点yeye,他的son结点叫baba,结点baba有两个结点:结点sunzei,结点sunnv。

本身Tarjan就是个DFS,所以搜索的话要一直处理完子树所有,也就是处理完结点baba的所有,才会处理到结点yeye,对于当前子树而言,并查集的作用也是对于当前子树的情况,所以结点sunzei和sunnv的LCA就是baba,不是yeye。

大致感觉也能感觉粗来吧?

PS:这份代码 HDU2586 改改就过了,而且题目中的说的空行是没有的~(蜜汁怂恿贴代码嫌疑。。)

#include <bits/stdc++.h>using namespace std;typedef long long LL;const int maxm=2e4+10;const int maxn=1e4+10;const int maxq=2e6+10;struct Node{int to;int w;int next;}e[maxm];int eh[maxn],dis[maxn],pre[maxn],etol,vis[maxn];struct Query{int to;int index;int next;}qe[maxq];int qh[maxn],ans[maxq/2],qtol;int n,m,c;void init(){etol=qtol=0;memset(eh,-1,sizeof(eh));memset(qh,-1,sizeof(qh));}void add1(int u,int v,int w){e[etol].to=v;e[etol].w=w;e[etol].next=eh[u];eh[u]=etol++;}void add2(int u,int v,int id){qe[qtol].index=id;qe[qtol].to=v;qe[qtol].next=qh[u];qh[u]=qtol++;}int Find(int u){if(pre[u]!=u) pre[u]=Find(pre[u]);return pre[u];}void LCA(int u,int deep,int root){pre[u]=u;dis[u]=deep;vis[u]=root;for(int i=eh[u];~i;i=e[i].next){int v=e[i].to;if(vis[v]==-1){LCA(v,deep+e[i].w,root);pre[v]=u;}}for(int i=qh[u];~i;i=qe[i].next){int v=qe[i].to;if(vis[v]==root)ans[qe[i].index]=dis[v]+dis[u]-2*dis[Find(v)];}}int main(){while(~scanf("%d%d%d",&n,&m,&c)){int u,v,w;init();while(m--){scanf("%d%d%d",&u,&v,&w);add1(u,v,w);add1(v,u,w);}for(int i=0;i<c;i++){scanf("%d%d",&u,&v);ans[i]=-1;add2(u,v,i);add2(v,u,i);}memset(vis,-1,sizeof(vis));for(int i=1;i<=n;i++){if(vis[i]==-1)LCA(i,0,i);}for(int i=0;i<c;i++){if(ans[i]==-1) puts("Not connected");else printf("%d\n",ans[i]);}}return 0;} 



0 0
原创粉丝点击