Educational Codeforces Round 7 - E. Ants in Leaves

题目链接：http://www.codeforces.com/contest/622/problem/E

题意是给你一棵树，1为根，每个叶子节点有一个蚂蚁，移动到一个邻接节点时间耗费为1，一个节点上不能同时有1个以上的蚂蚁数目（除了根节点外），让你求最后一个蚂蚁到达根节点的最少的时间。

可以先dfs预处理每个叶子节点的深度dep[i]，然后把1节点邻接的节点当作一个子根，求这个子根所在的子树上蚂蚁到这个子根的最大的时间，除了子树上蚂蚁最早到达的时间为ans[0] ，其余子树上的蚂蚁为max(ans[i - 1] , dep[i] + 1)，最后的答案就是遍历1所相邻的子根所得到的答案取一个最大值。

#include <bits/stdc++.h>
using namespace std;
const int MAXN = 5e5 + 5;
bool ok[MAXN];
vector <int> G[MAXN] , ans;

void dfs(int u , int par , int dep) {
if(G[u].size() == 1) {
ans.push_back(dep + 1);
return ;
}
for(int i = 0 ; i < G[u].size() ; i++) {
if(G[u][i] != par) {
dfs(G[u][i] , u , dep + 1);
}
}
}

int main()
{
int n , v , u;
scanf("%d" , &n);
for(int i = 1 ; i < n ; i++) {
scanf("%d %d" , &u , &v);
G[u].push_back(v);
G[v].push_back(u);
}
int len = G[1].size() , res = 0 , temp;
for(int i = 0 ; i < len ; i++) {
dfs(G[1][i] , 1 , 0);
sort(ans.begin() , ans.end());
temp = ans[0];
for(int j = 1 ; j < ans.size() ; j++) {
temp = max(temp + 1 , ans[j]);
}
res = max(res , temp);
ans.clear();
}
printf("%d\n" , res);
}