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;
}