D. Reset K Edges codeforces 1739D

Dashboard - Educational Codeforces Round 136 (Rated for Div. 2) - Codeforces

题目大意：有一棵n个点的树，1号点为根，每次操作可以删去一条边，并将该边连接的一个点连接到1号点上，最多操作k次，问最小的树的层数是多少

1<=n<=2e5;1<=k<=n-1

思路：因为n的范围是2e5，所以我们可以二分最小层数，每次二分遍历整棵树，按照最小层数的限制对树进行修改，然后比较实际进行的修改次数和限制次数k，但我们可以发现在下图中，如果我们从根开始向下统计每个点的深度，深度>限制深度就将点连接1，如果当前的限制深度为2，按照如上操作应该将3,4,6,8四个点连到1上，而实际上如果将7,9,5连到1上，操作次数少1，也就是我们从上向下遍历的时候，我们选择哪个点连接到1对后面的结果是有后效性的，所以无法考虑所有情况

,

所以我们的最优策略就是进行后序遍历，从下向上统计深度，如果当前点深度等于限制深度-1，就把这个点连到1上，同时该点深度变为-1，每个点在后序遍历时维护深度为当前深度和当前子节点深度+1最大值，这样就可以做到无后效性统计所有结果

#include<iostream>
#include<cstdio>
#include<queue>
#include<cmath>
using namespace std;
const int N = 2e5 + 5;
struct Edge
{
	int v, next;
}e[N * 2];
int head[N];
int cnt = 0;
int n;
int kk;
int d[N];
void init()
{//初始化变量
	for (int i = 1; i <= n; i++)
	{
		head[i] = -1;
		e[i].next = -1;
		e[i].v = 0;
		d[i] = 0;
	}
	cnt = 0;
}
void addedge(int u, int v)
{
	e[++cnt].v = v;
	e[cnt].next = head[u];
	head[u] = cnt;
}//链式前向星存图
void dfs(int u, int fa, int dee)
{//统计在当前限制深度下需要的操作次数
	for (int i = head[u]; ~i; i = e[i].next)
	{
		int v = e[i].v;
		dfs(v, u, dee);
		d[u] = max(d[u], d[v]+1);//在后续遍历时维护节点深度
	}
	if (d[u]  == dee - 1 && fa != 1 && u != 1)//注意节点1和节点1连接的点不能操作
	{
		kk++;//当前点等于限制深度-1就连到1上	
		d[u] = -1;//当前节点深度置为-1，便于维护父结点深度,
	}
	return;
}
int k;
bool check(int x)
{
	for (int i = 1; i <= n; i++)
		d[i] = 0;//每次二分都要重新统计深度
	kk = 0;//记录实际操作次数
	dfs(1, 0, x);//当前节点，父结点，限制操作次数
	return kk <= k;
}
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		scanf("%d%d", &n, &k);
		init();
		for (int i = 1; i <= n - 1; i++)
		{
			int u;
			scanf("%d", &u);
			addedge(u, i + 1);
		}
		int l = 1, r = n, mid = 1;
		int ans = 1;
		while (l <= r)
		{
			mid = (l + r) >> 1;
			if (check(mid))
			{
				r = mid - 1;//当前操作次数小于限制次数，层数可能更小
				ans = mid;
			}
			else
			{
				l = mid + 1;
			}
		}

		printf("%d\n", ans);
	}
	return 0;
}