京东2019笔试:紧急疏散
体育场突然着火了,现场需要紧急疏散,但是过道真的是太窄了,同时只能容许一个人通过。现在知道了体育场的所有座位分布,座位分布图是一棵树,已知每个座位上都坐了一个人,安全出口在树的根部,也就是1号结点的位置上。其他节点上的人每秒都能向树根部前进一个结点,但是除了安全出口以外,没有任何一个结点可以同时容纳两个及以上的人,这就需要一种策略,来使得人群尽快疏散,问在采取最优策略的情况下,体育场最快可以在多长时间内疏散完成。
输入描述:第一行包含一个正整数n,即树的结点数量(1<=n<=100000)。 接下来有n-1行,每行有两个正整数x,y,表示在x和y结点之间存在一条边。(1<=x,y<=n)
输出描述:输出仅包含一个正整数,表示所需要的最短时间
例子:
// 输入
6
2 1
3 2
4 3
5 2
6 1
// 输出
4
图示,最快速的策略用时取决于节点 1 的左右子树中节点的数量。采用深度优先遍历,计算树的节点数量。
邻接表表示图结构,包括头节点和表节点(边界点),头节点的ptr指向从该头节点出发的所有边的第一个边节点,其他所有边有表节点的next指针链接。不好写但是效率高。
链式向前星
其实就是静态建立的邻接表,用边集数组和邻接表相结合,可以快速访问一个顶点的所有邻接点,时间效率为O(m),空间效率也为O(m),遍历效率也为O(m)。和邻接表定义类似,包含两个结构:
边集数组:edge[ ],edge[i]表示第 i 条边;
头结点数组:head[ ],head[i]存以 i 为起点的第一条边的下标(在edge[ ]中的下标)
const int N=1e+10, M=N*2;
struct Edge {
int to, next;
} edge[M];
int Head[N], idx, n;
void add(int a, int b)
{
edge[idx].to = b;
edge[idx].next = head[a];
Head[a] = idx++; // 题目中节点为1...n
}
建立起链式向前星:
memset(Head, -1, sizeof(Head)); // 头文件#include<cstring>
for(int i = 0;i<n-1;++i){
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
深度优先遍历:注意深度最后不能计算回根节点
int dfs(int u, int fa)
{
int res = 1;
for(int i=Head[u];~i;i=edge[i].next){
if (edge[i].to!=fa) res+=dfs(edge[i].to, u);
}
return res;
}
全部代码
#include <iostream>
#include <cstring>
using namespace std;
const int N = 1e5 + 10, M = N * 2;
struct Edge {
int to, next;
} edge[M];
int Head[N], idx, n;
void add(int a, int b)
{
edge[idx].to = b;
edge[idx].next = head[a];
Head[a] = idx++;
}
int dfs(int u, int fa)
{
int res = 1;
for (int i = head[u]; ~i; i = edge[i].next)
{
if (edge[i].to != fa) res += dfs(edge[i].to, u);
}
return res;
}
int main()
{
cin >> n;
memset(Head, -1, sizeof(Head));
for (int i = 0; i < n - 1; ++i)
{
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
int res = 0;
for (int i = Head[1]; ~i; i = edge[i].next)
res = max(res, dfs(edge[i].to, 1));
cout << res << endl;
return 0;
}