BZOJ3835 : [Poi2014]Supercomputer

设h为树的高度，sum[i]为深度大于i的点的个数

则$ans=\max(i+\lceil\frac{sum[i]}{k}\rceil)$

注意到这是一条直线，于是斜率优化至$O(n)$

 

#include<cstdio>
#define N 1000010
typedef long long ll;
int n,m,i,x,g[N],nxt[N],v[N],ed,Q[N],c[N],sum[N],d,q[N],h=1,t,ans[N];
inline void read(int&a){char ch;while(!(((ch=getchar())>='0')&&(ch<='9')));a=ch-'0';while(((ch=getchar())>='0')&&(ch<='9'))a*=10,a+=ch-'0';}
inline void add(int x,int y){v[++ed]=y;nxt[ed]=g[x];g[x]=ed;}
void dfs(int x,int d){c[d++]++;for(int i=g[x];i;i=nxt[i])dfs(v[i],d);}
inline int ceil(int a,int b){return a/b+(a%b>0);}
int main(){
  read(n),read(m);
  for(i=1;i<=m;i++)read(Q[i]);
  for(i=2;i<=n;i++)read(x),add(x,i);
  dfs(1,1);
  while(c[++d]);
  for(i=--d;~i;i--)sum[i]=sum[i+1]+c[i+1];
  for(i=d;~i;q[++t]=i--)while(h<t&&(ll)(q[t-1]-q[t])*(sum[i]-sum[q[t]])>=(ll)(q[t]-i)*(sum[q[t]]-sum[q[t-1]]))t--;
  for(i=n;i;i--){
    while(h<t&&(ll)i*(q[h]-q[h+1])<=sum[q[h+1]]-sum[q[h]])h++;
    ans[i]=q[h]+ceil(sum[q[h]],i);
  }
  for(i=1;i<=m;i++)printf(i<m?"%d ":"%d",Q[i]>n?d:ans[Q[i]]);
  return 0;
}