CodeVS 2370 LCA 解题报告

xiaoxiao2021-02-28  99

2370 小机房的树

题目描述 Description

小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力

输入描述 Input Description

第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。 第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点 输出描述 Output Description 一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。

样例输入 Sample Input

3 1 0 1 2 0 1 3 1 0 2 0 1 2

样例输出 Sample Output

1 1 2

数据范围及提示 Data Size & Hint

1<=n<=50000, 1<=m<=75000, 0<=c<=1000

【解题报告】 倍增求LCA裸题。 莫名写挂,调了半天。。。。。

代码如下:

#include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define N 50010 #define M 75010 int n,m,cnt=0; int dis[N],deep[N],head[M],fa[N][30]; struct Edge { int to,nxt,w; }e[M<<1]; void adde(int u,int v,int w) { e[++cnt].w=w; e[cnt].to=v; e[cnt].nxt=head[u]; head[u]=cnt; } void dfs(int u,int x) { deep[u]=x; for(int i=head[u];~i;i=e[i].nxt) { int v=e[i].to; if(!deep[v]&&v) { fa[v][0]=u; dis[v]=dis[u]+e[i].w; dfs(v,x+1); } } } void getfa() { for(int i=0;i<n;++i) for(int j=1;j<=22;++j) fa[i][j]=fa[fa[i][j-1]][j-1]; } int get_same(int u,int v) { for(int i=0;i<=22;i++) if((1<<i)&v) u=fa[u][i]; return u; } int lca(int u,int v) { if(deep[u]<deep[v]) swap(u,v); u=get_same(u,deep[u]-deep[v]); for(int i=22;i>=0;i--) if(fa[u][i]!=fa[v][i]) { u=fa[u][i]; v=fa[v][i]; } if(u==v) return u; return fa[u][0]; } int main() { memset(head,-1,sizeof(head)); scanf("%d",&n); for(int i=1;i<n;++i) { int u,v,w;scanf("%d%d%d",&u,&v,&w); adde(u,v,w); adde(v,u,w); } dfs(0,0); getfa(); scanf("%d",&m); for(int i=1;i<=m;++i) { int u,v;scanf("%d%d",&u,&v); int ans=lca(u,v); if(ans==u) printf("%d\n",dis[v]-dis[u]); else printf("%d\n",dis[u]+dis[v]-2*dis[ans]); } return 0; }

板子:

#include<cstdio> #include<cstring> #include<algorithm> using namespace std; #define N 50010 int t,n,m,cnt=-1,head[N]; struct Edge{int to,nxt,w;}e[N<<1]; int dis[N],dep[N],fa[N],anc[N][22]; void adde(int u,int v,int w) { e[++cnt].to=v;e[cnt].w=w; e[cnt].nxt=head[u];head[u]=cnt; e[++cnt].to=u;e[cnt].w=w; e[cnt].nxt=head[v];head[v]=cnt; } void dfs(int u) { anc[u][0]=fa[u]; for(int i=1;i<=20;++i) anc[u][i]=anc[anc[u][i-1]][i-1]; for(int i=head[u];~i;i=e[i].nxt) { int v=e[i].to; if(v==fa[u]) continue; fa[v]=u; dep[v]=dep[u]+1; dis[v]=dis[u]+e[i].w; dfs(v); } } int lca(int u,int v) { if(dep[u]<dep[v]) swap(u,v); for(int i=20;i>=0;--i) { if(dep[anc[u][i]]>=dep[v]) { u=anc[u][i]; } } if(u==v) return u; for(int i=20;i>=0;--i) { if(anc[u][i]!=anc[v][i]) { u=anc[u][i]; v=anc[v][i]; } } return anc[u][0]; } int main() { for(scanf("%d",&t);t;--t) { cnt=-1; memset(head,-1,sizeof(head)); memset(fa,0,sizeof(fa)); memset(anc,0,sizeof(anc)); memset(dep,0,sizeof(dep)); memset(dis,0,sizeof(dis)); scanf("%d%d",&n,&m); for(int i=1;i<n;++i) { int u,v,w;scanf("%d%d%d",&u,&v,&w); adde(u,v,w); } dfs(1); // for(int i=1;i<=n;++i) printf("%d ",dep[i]);puts(""); // for(int i=1;i<=n;++i) printf("%d ",dis[i]);puts(""); for(int i=1;i<=m;++i) { int u,v;scanf("%d%d",&u,&v); printf("%d\n",dis[u]+dis[v]-2*dis[lca(u,v)]); } } return 0; } /* 1 5 1 1 2 10 1 3 10 3 4 10 3 5 10 4 2 */
转载请注明原文地址: https://www.6miu.com/read-30258.html

最新回复(0)