E. Ants in Leaves

题意
有一颗树,树的每个叶子上都有一个蚂蚁,蚂蚁往树根处爬。其中每个结点(根节点除外)最多只能出现一只蚂蚁。问最少经过多长时间全部的蚂蚁才能都到达根节点
思路
由于只有根节点可以同时有多只蚂蚁,so 我们先将根节点的子节点拆除。这样就将一棵树转化为森林,显然森林的其中的一颗耗时最长的树就是最后的答案。
所以我们的问题就是如何计算每棵树最长的耗时。

对于其中的一棵树, 先dfs处理出所有结点的深度,然后对深度排序, 然后计算每个叶子结点到达根结点的时间。 假设第i个蚂蚁到达根结点的时间等于a[i],那么显然a[i] = max(a[i-1]+1, d[i]),等于上一个蚂蚁的时间+1和它所在深度的最大值。

#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
#include 
using namespace std;
#define L(i) i<<1
#define R(i) i<<1|1
#define INF  0x3f3f3f3f
#define pi acos(-1.0)
#define eps 1e-3
#define maxn 1000100
#define MOD 1000000007
#define inf 0x3f3f3f3f3f3f3f3f
struct Edge
{
    int to,next;
}edge[maxn];
struct node
{
    int dep,pos;
    bool operator<(const node &a)const
    {
        return dep < a.dep;
    }
}st[maxn];
int n,m;
int dp[maxn];
int tot,head[maxn];
void add_edge(int u,int v)
{
    edge[tot].to = v;
    edge[tot].next = head[u];
    head[u] = tot++;
}
void dfs(int u,int pre,int dep,int &len)
{
    if(edge[head[u]].to == pre && edge[head[u]].next == -1)
    {
        dp[len++] = dep;
        return;
    }
    for(int i = head[u]; i != -1; i = edge[i].next)
    {
        int v = edge[i].to;
        if(v == pre)
            continue;
        dfs(v,u,dep+1,len);
    }
}
int main()
{
    int t,C = 1;
    //scanf("%d",&t);
    while(scanf("%d",&n) != EOF)
    {
        tot = 0;
        memset(head,-1,sizeof(head));
        int x,y;
        for(int i = 0; i < n-1; i++)
        {
            scanf("%d%d",&x,&y);
            add_edge(x,y);
            add_edge(y,x);
        }
        int ans = 0;
        dp[0] = 0;
        for(int i = head[1]; i != -1; i = edge[i].next)
        {
            int len = 1;
            dfs(edge[i].to,1,1,len);
            sort(dp+1,dp+len);
            for(int i = 1; i < len; i++)
                dp[i] = max(dp[i],dp[i-1]+1);
            ans = max(ans,dp[len-1]);
        }
        printf("%d\n",ans);
    }
    return 0;
}



你可能感兴趣的:(搜索—DFS)