Description
一棵n个节点的有向树,以1为根,每个叶子节点有一只蚂蚁,蚂蚁一秒会往上爬一层,但是除了根节点一个节点最多只能有一只蚂蚁,问最少经过多少秒所有蚂蚁可以到达根节点
Input
第一行一整数n表示点数,之后n-1行每行两个整数u和v表示树上一条边(2<=n<=5e5)
Output
输出所有蚂蚁到达根节点的最少用时
Sample Input
12
1 2
1 3
1 4
2 5
2 6
3 7
3 8
3 9
8 10
8 11
8 12
Sample Output
6
Solution
考虑1节点的儿子节点,一个儿子节点每秒最多向根节点送去一只蚂蚁,设以这个儿子节点为根的子树中各叶子节点的深度分别为d[1],d[2],…,d[m](升序排,一遍dfs可得到),那么第一只到达根节点的蚂蚁用时为d[1],而第二只蚂蚁有两种情况,若d[2]=d[1],那么这只蚂蚁只能等一秒等上一只蚂蚁到根之后它再走,所以第二只蚂蚁在d[2]+1秒到达根节点;若d[2]>d[1],那么这只蚂蚁不用等,在d[2]秒到达根节点,故d[i]=max(d[i-1]+1,d[i])为第i只蚂蚁到达根节点的时间,那么d[m]为这个儿子节点为根的子树中所有蚂蚁到达根节点的用时,从所有儿子节点中选一个最大值即为答案
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
#define INF 0x3f3f3f3f
#define maxn 500005
int n;
vector<int>g[maxn],d;
void dfs(int u,int fa,int deep)
{
if(g[u].size()==1)d.push_back(deep);
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(v==fa)continue;
dfs(v,u,deep+1);
}
}
int main()
{
while(~scanf("%d",&n))
{
for(int i=1;i<=n;i++)g[i].clear();
for(int i=1;i<n;i++)
{
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v),g[v].push_back(u);
}
int ans=0;
for(int k=0;k<g[1].size();k++)
{
d.clear();
dfs(g[1][k],1,1);
sort(d.begin(),d.end());
for(int i=1;i<d.size();i++)
d[i]=max(d[i-1]+1,d[i]);
ans=max(ans,d.back());
}
printf("%d\n",ans);
}
return 0;
}