codeforces 622E. Ants in Leaves

题目链接

给一棵有根树， 每个叶子节点上有一只蚂蚁。 在0时刻蚂蚁开始向上爬， 同一时刻， 除了根节点以外， 一个节点上面不能有2个蚂蚁。 问所有的蚂蚁都爬到根节点需要的最短时间。

因为除了根节点， 一个节点上面只能有一个蚂蚁， 所以我们将根节点去掉， 于是就有了一个森林。 时间就是所有子树里面花费时间最多的那棵树的时间。 

对于每棵子树， 我们将所有的节点标一个深度， 然后将深度排序。 可以知道 dp[i] = max(dp[i-1]+1, d[i])， d是深度， 这样就可以求得最终答案。

#include <iostream>
#include <vector>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <string>
#include <queue>
#include <stack>
#include <bitset>
using namespace std;
#define pb(x) push_back(x)
#define ll long long
#define mk(x, y) make_pair(x, y)
#define lson l, m, rt<<1
#define mem(a) memset(a, 0, sizeof(a))
#define rson m+1, r, rt<<1|1
#define mem1(a) memset(a, -1, sizeof(a))
#define mem2(a) memset(a, 0x3f, sizeof(a))
#define rep(i, n, a) for(int i = a; i<n; i++)
#define fi first
#define se second
typedef pair<int, int> pll;
const double PI = acos(-1.0);
const double eps = 1e-8;
const int mod = 1e9+7;
const int inf = 1061109567;
const int dir[][2] = { {-1, 0}, {1, 0}, {0, -1}, {0, 1} };
const int maxn = 5e5+5;
int head[maxn*2], num, cnt, dp[maxn], d[maxn];
struct node
{
    int to, nextt, w;
}e[maxn*2];
void add(int u, int v) {
    e[num].to = v, e[num].nextt = head[u], head[u] = num++;
}
void init() {
    num = 0;
    mem1(head);
}
void dfs(int u, int fa, int depth) {
    int flag = 0;
    for(int i = head[u]; ~i; i = e[i].nextt) {
        int v = e[i].to;
        if(v == fa)
            continue;
        flag = 1;
        dfs(v, u, depth+1);
    }
    if(!flag) {
        d[cnt++] = depth;
    }
}
int main()
{
    init();
    int n, u, v, ans = 0;
    cin>>n;
    for(int i = 1; i<n; i++) {
        scanf("%d%d", &u, &v);
        add(u, v);
        add(v, u);
    }
    for(int i = head[1]; ~i; i = e[i].nextt) {
        int v = e[i].to;
        cnt = 0;
        dfs(v, 1, 1);
        sort(d, d+cnt);
        for(int j = 1; j<cnt; j++) {
            d[j] = max(d[j-1]+1, d[j]);
        }
        ans = max(ans, d[cnt-1]);
    }
    cout<<ans<<endl;
    return 0;
}