（2020牛客暑期多校训练营）[四] Ancient Distance

原题
题目描述
给定N个点构成的有根树，顶点编号从1-N，根节点为1号点。你可以分配最多K个关节点，使得所有点的最大“祖先距离”尽可能的小。点x的“祖先距离”是在点x到根节点上的路径上，点x与第一个关键点的距离。若x到根节点的路径上没有关键点，则为正无穷大。例如1-2-3树上，关键点为2，则三个点的“祖先距离”分别为{+∞，0，1}。
（给一个有根树，在树上选择 k 个关键点(根必须选)，最小化点到最近关键祖先距离的最大值，求出 k 分别为 1,2,…,n 时答案的和）
Input
多组数据
输入N
在第二行中有N-1个整数第i个整数f[i]表示f[i]与i+1之间存在一条边
Output
输出 k 分别为 1,2,…,n 时答案的和
样例
输入

3
1 2
3
1 1

输出

3
2

思路
这道题求的是最大值的最小值，所以我们会很容易地想到二分，二分枚举k。知道k后，我们还要尽可能优地放好关键点。脑袋一拍我们会知道，关键点要尽量往上放，但是又不能太往上。根据这个策略，我们可以总结出，要把关键点放在深度最深的节点向上k个点的节点上。而且，每放完一个，就要把它的子树删掉。但是最后我们又要用这个节点，所以，我们就要用标记的方法解决。但是，我们现在还没有树！因此，要在开始时把树建好，这就用dfs序和线段树维护一下就可以了。
总的来说就是二分答案，贪心找最深的点，向上k个点放关键点，删除其子树，至所有节点都被删除。
时间复杂度O(n$lo{g}_n$$lo{g}_n$$lo{g}_n$)
代码

#include<bits/stdc++.h>
using namespace std;
const int maxn=2e5+5;
vector<int>vec[maxn];long long sum;
int n,cnt,mx,x,b[maxn],c[maxn],d[maxn],p[maxn],q[maxn];
inline int rd()
{
    int x=0,f=1;char c=0;
    while(!isdigit(c)){if(c=='-')f=-1;c=getchar();}
    while(isdigit(c))x=(x<<3)+(x<<1)+(c-'0'),c=getchar();
    return x*f;
}
void dfs(int x,int y)
{
    b[++cnt]=x,c[x]=y,d[x]=d[y]+1;
	for(int i=0;i<vec[x].size();i++)if(vec[x][i]!=y)dfs(vec[x][i],x);
}
int find(int w)
{
	int num=0;
	for(int i=1;i<=n;i++)q[i]=d[i];
	for(int i=n;i>=1;i--)
	{
		if(i==1||q[b[i]]-d[b[i]]==w) num++,q[b[i]]=-1;
		q[c[b[i]]]=max(q[c[b[i]]],q[b[i]]);
	}
	return num;
}
void dichotomia(int a,int b,int left,int right)
{
	if(a>b || left>right) return;
	if(left==right) {for(int i=a;i<=b;i++) p[i]=left;return;}
	int mid=(a+b)/2;p[mid]=find(mid);
	dichotomia(a,mid-1,p[mid],right);
	dichotomia(mid+1,b,left,p[mid]);
}
int main()
{
	while(~scanf("%d",&n))
	{
		for(int i=1;i<=n;i++)vec[i].clear();
		for(int i=2;i<=n;i++){x=rd();vec[x].push_back(i);vec[i].push_back(x);}
		d[0]=-1;cnt=mx=sum=0;dfs(1,0);
		for(int i=1;i<=n;i++)mx=max(mx,d[i]);
		dichotomia(0,mx,1,n);
		for(int i=1;i<=mx;i++) sum+=1ll*i*(p[i-1]-p[i]);
		printf("%lld\n",sum);
	}
	return 0;
}