CodeForces 622 E.Ants in Leaves（dfs）

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
#include
#include
#include
#include
#include
#include
#include
#include
#include
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;iint 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;iint u,v;
            scanf("%d%d",&u,&v);
            g[u].push_back(v),g[v].push_back(u); 
        }
        int ans=0;
        for(int k=0;k1].size();k++)
        {
            d.clear();
            dfs(g[1][k],1,1);   
            sort(d.begin(),d.end());
            for(int i=1;i1]+1,d[i]);
            ans=max(ans,d.back());
        }
        printf("%d\n",ans);
    }
    return 0;
}