首页 > 代码库 > POJ 1986 Distance Queries(LCA)
POJ 1986 Distance Queries(LCA)
【题目链接】 http://poj.org/problem?id=1986
【题目大意】
给出一棵树,问任意两点间距离。
【题解】
u,v之间距离为dis[u]+dis[v]-2*dis[LCA(u,v)]
【代码】
#include <cstdio>#include <algorithm>#include <cstring>using namespace std;const int N=300010;int d[N],num[N],dis[N],ed=0,x,y,c,n,m,i,w[N],v[N],vis[N],f[N],g[N],nxt[N],size[N],son[N],st[N],en[N],dfn,top[N],t;char ch;void read(int&a){char c;while(!(((c=getchar())>=‘0‘)&&(c<=‘9‘)));a=c-‘0‘;while(((c=getchar())>=‘0‘)&&(c<=‘9‘))(a*=10)+=c-‘0‘;}void add_edge(int x,int y,int _w){v[++ed]=y;w[ed]=_w;nxt[ed]=g[x];g[x]=ed;}void dfs(int x){ size[x]=1; for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){ f[v[i]]=x,d[v[i]]=d[x]+1;dis[v[i]]=dis[x]+w[i]; dfs(v[i]),size[x]+=size[v[i]]; if(size[v[i]]>size[son[x]])son[x]=v[i]; }}void dfs2(int x,int y){ st[x]=++dfn;top[x]=y; if(son[x])dfs2(son[x],y); for(int i=g[x];i;i=nxt[i])if(v[i]!=son[x]&&v[i]!=f[x])dfs2(v[i],v[i]); en[x]=dfn;}int lca(int x,int y){ for(;top[x]!=top[y];x=f[top[x]])if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;} return d[x]<d[y]?x:y;}void init(){ memset(g,dfn=ed=0,sizeof(g)); memset(v,0,sizeof(v)); memset(nxt,0,sizeof(nxt)); memset(son,-1,sizeof(son));}int main(){ while(~scanf("%d%d",&n,&m)){ init(); while(m--){ int u,v,w; char op[10]; scanf("%d%d%d%s",&u,&v,&w,op); add_edge(u,v,w); add_edge(v,u,w); }dfs(1);dfs2(1,1); scanf("%d",&m); while(m--){ int u,v; scanf("%d%d",&u,&v); printf("%d\n",dis[u]+dis[v]-2*dis[lca(u,v)]); } }return 0;}
POJ 1986 Distance Queries(LCA)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。