洛谷 4364 [九省联考2018]IIIDX——“预留”的思路

题目：https://www.luogu.org/problemnew/show/P4364

原来想了一个错误的思路，就是这样：

　　solve( cr , l , r ) 表示 cr 为根的子树填 [ l , r ] 的数；然后把 l 给 cr ，剩下的 [ l+1 , r ] 分成一段一段，大的段给标号小的孩子。

然后只能得 60 分。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
#define db double
#define pb push_back
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
const int N=5e5+;
int n,d[N],p[N],siz[N]; db k;
vector<int> vt[N];
void dfs(int cr)
{
  siz[cr]=;
  for(int i=,lm=vt[cr].size();i<lm;i++)
    dfs(vt[cr][i]), siz[cr]+=siz[vt[cr][i]];
}
void solve(int cr,int l,int r)
{
  p[cr]=d[l]; int p0=l;
  for(int i=vt[cr].size()-;i>=;i--)
    {
      solve(vt[cr][i],p0+,p0+siz[vt[cr][i]]);
      p0+=siz[vt[cr][i]];
    }
}
int main()
{
  n=rdn();scanf("%lf",&k);
  for(int i=;i<=n;i++)d[i]=rdn();
  sort(d+,d+n+);
  for(int i=;i<=n;i++)
    vt[(int)floor(i/k)].pb(i);
  dfs(); solve(,,n);
  for(int i=;i<=n;i++)printf("%d ",p[i]);
  puts(""); return ;
}

看了看题解。https://files-cdn.cnblogs.com/files/NaVi-Awson/IIIDX.pdf

关于 “去掉父亲预留的影响” ，是这样考虑：

　　按顺序枚举节点，在树上就是像 bfs 一样遍历；

　　一个点 cr 要预留，是为了让 “和它深度相同的点” 不要抢了它的位置。

　　当开始遍历 cr 的子树的时候，直接去掉 cr 的影响并再也不加入了；这样不会有错，因为此时不会再遍历和 cr 深度相同的点了；

　　下一层的点已经由它们的父亲留好了位置，所以去掉 cr 的影响也不会让之后的点出错。

还有一种不太能理解的做法：https://www.cnblogs.com/HocRiser/p/8742680.html

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
#define ls Ls[cr]
#define rs Rs[cr]
using namespace std;
int rdn()
{
  int ret=;bool fx=;char ch=getchar();
  while(ch>''||ch<''){if(ch=='-')fx=;ch=getchar();}
  while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
  return fx?ret:-ret;
}
int Mn(int a,int b){return a<b?a:b;}
int Mx(int a,int b){return a>b?a:b;}
const int N=5e5+,M=N<<;
int n,a[N],fa[N],ans[N],siz[N],tp[N],tx[N];
int tot,Ls[M],Rs[M],mn[M],tg[M];
void pshp(int cr){mn[cr]=Mn(mn[ls],mn[rs]);}
void build(int l,int r,int cr)
{
  if(l==r){mn[cr]=n-l;return;}
  int mid=l+r>>;
  ls=++tot; build(l,mid,ls);
  rs=++tot; build(mid+,r,rs);
  pshp(cr);
}
void pshd(int cr)
{
  if(!tg[cr])return; int w=tg[cr]; tg[cr]=;
  tg[ls]+=w; tg[rs]+=w; mn[ls]+=w; mn[rs]+=w;
}
void mdfy(int l,int r,int cr,int L,int R,int k)
{
  if(l>=L&&r<=R){tg[cr]+=k;mn[cr]+=k;return;}
  int mid=l+r>>; pshd(cr);
  if(L<=mid)mdfy(l,mid,ls,L,R,k);
  if(mid<R)mdfy(mid+,r,rs,L,R,k);
  pshp(cr);
}
int qry(int l,int r,int cr,int k)
{
  if(l==r)return mn[cr]>=k?l:-;
  int mid=l+r>>,ret=-; pshd(cr);
  if(mn[ls]>=k)ret=qry(mid+,r,rs,k);
  if(ret>=)return ret;
  return qry(l,mid,ls,k);
}
int main()
{
  n=rdn(); double sl; scanf("%lf",&sl);
  for(int i=;i<=n;i++)a[i]=rdn();
  sort(a+,a+n+);
  for(int i=;i<=n;i++)tp[i]=a[i];
  int m=unique(tp+,tp+n+)-tp-;
  for(int i=;i<=n;i++)
    {
      a[i]=lower_bound(tp+,tp+m+,a[i])-tp;
      tx[a[i]]++;
    }
  for(int i=;i<=m;i++)tx[i]+=tx[i-];
  for(int i=;i<=n;i++) fa[i]=floor(i/sl);
  for(int i=n;i;i--) siz[i]++,siz[fa[i]]+=siz[i];
  tot=; build(,n,);
  for(int i=;i<=n;i++)
    {
      if(i>&&fa[i]!=fa[i-])
    mdfy(,n,,,ans[fa[i]]-,siz[fa[i]]-);
      //-1 for already use
      int p=qry(,n,,siz[i])+;
      p=++tx[a[p]-];//
      ans[i]=p;
      mdfy(,n,,,p-,-siz[i]);
    }
  for(int i=;i<=n;i++)printf("%d ",tp[a[ans[i]]]);
  puts(""); return ;
}