C++之路进阶——codevs1036(商务旅行)

1036 商务旅行

题目描述 Description

某首都城市的商人要经常到各城镇去做生意,他们按自己的路线去做,目的是为了更好的节约时间。

假设有N个城镇,首都编号为1,商人从首都出发,其他各城镇之间都有道路连接,任意两个城镇之间如果有直连道路,在他们之间行驶需要花费单位时间。该国公路网络发达,从首都出发能到达任意一个城镇,并且公路网络不会存在环。

你的任务是帮助该商人计算一下他的最短旅行时间。

输入描述 Input Description

输入文件中的第一行有一个整数N,1<=n<=30 000,为城镇的数目。下面N-1行,每行由两个整数a 和b (1<=ab<=n; a<>b)组成,表示城镇a和城镇b有公路连接。在第N+1行为一个整数M,下面的M行,每行有该商人需要顺次经过的各城镇编号。

输出描述 Output Description

在输出文件中输出该商人旅行的最短时间。

样例输入 Sample Input
5
1 2
1 5
3 5
4 5
4
1
3
2
5
样例输出 Sample Output

7

思路:

裸lca。。。。,模板题。不懂看代码。

 #include<cstdio>
#include<iostream>
#include<cmath>
#include<vector>
#include<algorithm>
#define maxn 30010 using namespace std; vector<int> s[];
int fa[maxn][],deep[maxn],n,m; void getdeep(int k,int d)
{
deep[k]=d;
for (int i=;(<<i)<=deep[k];i++)
fa[k][i]=fa[fa[k][i-]][i-];
for (int i=;i<s[k].size();i++)
{
if (!deep[s[k][i]])
{
fa[s[k][i]][]=k;
getdeep(s[k][i],d+);
}
}
} int lca(int x,int y)
{
if (deep[x]<deep[y]) swap(x,y);
int t=deep[x]-deep[y];
for (int i=;i<;i++)
if ((<<i)&t) x=fa[x][i];
for (int i=;i>=;i--)
{
if (fa[x][i]!=fa[y][i])
{
x=fa[x][i];
y=fa[y][i];
}
}
if (x==y) return x;
return fa[x][];
} int main()
{
scanf("%d",&n);
for (int i=;i<n-;i++)
{
int x,y;
scanf("%d%d",&x,&y);
s[y].push_back(x);
s[x].push_back(y);
}
getdeep(,);
scanf("%d",&m);
int anss=;
int a=,b;
for (int i=;i<m;i++)
{
scanf("%d",&b);
anss+=deep[a]+deep[b]-*deep[lca(a,b)];
a=b;
}
printf("%d",anss);
return ;
}
上一篇:Python爬虫学习(7):浙大软院网号嗅探


下一篇:线上mongodb 数据库用户到期时间修改的操作记录