Gym - 101522B树的直径

Gym - 101522B

题意：

​ 给出一棵树，然后按照所给出的方式建立新的边，然后可以一层层的建边直到无边可建。问需要多少个小时，每一个小时可以建立一条边，可以多线程建立。

思路：

​ 画图可以找到规律，最坏的情况是n个点是一个线性的，那么时间将是最长，根据线性找规律发现其长度与时间是以2倍递增的，每一个新的小时便可以在原来的基础上画2倍的线，所以找出最长边然后判断2的ans次方即可。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>

using namespace std;

const int maxn = 5e5+10;

struct  Edge
{
    int to,v;
    int next;
}edge[maxn*2];

int n;
int head[maxn],pos;

int dp[maxn][3];
int sonMax[maxn];

void init()
{
    pos = 0;
    memset(head,-1,sizeof(head));
    memset(dp,0,sizeof(dp));
    memset(sonMax,0,sizeof(sonMax));
}

void add_edge(int u,int v,int val)
{
    edge[pos].to = v;
    edge[pos].v = val;
    edge[pos].next = head[u];
    head[u] = pos++;
    edge[pos].to = u;
    edge[pos].v = val;
    edge[pos].next = head[v];
    head[v] = pos++;
}

void input()
{
    for(int i = 2;i <= n; i++) {
        int a,b;
        scanf("%d%d",&a,&b);
        add_edge(a,b,1);
    }
}

void dfs1(int u,int fa)
{
    dp[u][0] = dp[u][1] = dp[u][2] = sonMax[u] = 0;
    for(int i = head[u];i != -1; i = edge[i].next) {
        int to = edge[i].to;
        if(to == fa) continue;
        dfs1(to,u);
        if(dp[u][0] < dp[to][0] + edge[i].v) {
            sonMax[u] = to;
            dp[u][1] = max(dp[u][0],dp[u][1]);
            dp[u][0] = dp[to][0] + edge[i].v;
        }
        else dp[u][1] = max(dp[u][1],dp[to][0] + edge[i].v);   
    }
}

void dfs2(int u,int fa)
{
    for(int i = head[u];i != -1; i = edge[i].next) {
        int to = edge[i].to;
        if(to == fa) continue;
        if(sonMax[u] == to) {
            dp[to][2] = max(dp[u][2],dp[u][1]) + edge[i].v;
        }
        else
            dp[to][2] = max(dp[u][2],dp[u][0]) + edge[i].v;
        dfs2(to,u);
    }
}

int main(int argc, char const *argv[])
{
    //freopen("in.txt","r",stdin);

    scanf("%d",&n);
    init();
    input();
    dfs1(1,-1);
    dfs2(1,-1);
    int Max = 0;
    for(int i = 1;i <= n; i++) {
        Max = max(Max,dp[i][0]+dp[i][2]);
    }
    long long two = 2;
    int ans = 1;
    while(two < Max)  {
        two = two*2;
        ans++;
    }
    if(n == 2) ans = 0;
    printf("%d\n",ans);
    return 0;
}