首页 > 代码库 > hdu2586 LCA
hdu2586 LCA
给一颗无向树有n个结点,有m个询问,问树上任意两点间距离,n的范围是40000,m是200
这题告诉我们一个求树上两点间距离的好方法,就是先求根到其余所有点的距离,再求出询问的LCA,答案为dis[a]+dis[b]-2*dis[lca(a,b)]
#include <iostream> #include <cstdlib> #include <cstring> #include <string> #include <cstdio> #include <cmath> #include <algorithm> #include <vector> #include <queue> #include <map> #define inf 0x3f3f3f3f #pragma comment(linker, "/STACK:16777216") #define eps 1e-6 #define ll long long using namespace std; const int maxn=40010; int dis[maxn]; struct node { int v,id,w;//id为边号或询问号 node* next; }ed[maxn<<2],*head[maxn],*q[maxn]; struct qnode { int u,v,ans;//存询问结点,ans最近公共祖先 } qu[maxn]; int fa[maxn],vis[maxn],cnt; void init(int n) { cnt=0; memset(vis,0,sizeof vis); memset(head,0,sizeof head); memset(q,0,sizeof q); for(int i=0;i<=n;i++) fa[i]=i; } int getfa(int x) { if(fa[x]==x) return x; return fa[x]=getfa(fa[x]); } void LCA(int u) { fa[u]=u,vis[u]=1; for(node *p=q[u];p!=NULL;p=p->next) { if(vis[p->v]) { int id=p->id; qu[id].ans=getfa(p->v); } } for(node *p=head[u];p!=NULL;p=p->next) { if(!vis[p->v]) { LCA(p->v); fa[p->v]=u; } } } void adde(node *e[],int u,int v,int w,int id)//建边e传头节点数组。为询问边的或树边的。 { ed[cnt].v=v; ed[cnt].w=w; ed[cnt].id=id; ed[cnt].next=e[u]; e[u]=&ed[cnt++]; } void dfs(int s,int d) { for(node *p=head[s];p!=NULL;p=p->next) { if(!vis[p->v]) { vis[p->v]=1; dis[p->v]=d+(p->w); dfs(p->v,dis[p->v]); } } } int main() { int icy,a,b,c,i,m,n; scanf("%d",&icy); while(icy--) { scanf("%d%d",&n,&m); init(n); for(i=1;i<=n-1;i++) { scanf("%d%d%d",&a,&b,&c); adde(head,a,b,c,i); adde(head,b,a,c,i); } for(i=1;i<=m;i++) { scanf("%d%d",&a,&b); adde(q,a,b,0,i); adde(q,b,a,0,i); qu[i].u=a; qu[i].v=b; } LCA(1); memset(dis,0,sizeof dis); memset(vis,0,sizeof vis); vis[1]=1;dis[1]=0; dfs(1,0); for(i=1;i<=m;i++) printf("%d\n",dis[qu[i].u]+dis[qu[i].v]-dis[qu[i].ans]-dis[qu[i].ans]); } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。