蓝书（算法竞赛进阶指南）刷题记录——POJ2182 Lost Cows（树状数组+倍增）

题目：POJ2182.
题目大意：给定$a[i]$表示第$i$个位置前面有多少个数比它小，求一个长度为$n$的排列.
$1\le n\le 8000$.

考虑倒着处理这个问题，容易发现一个位置的值就是$[1,n]$去掉它后面的值后的第$a[i]$小.

考虑用一个$01$序列的角度来看这个东西，其实就是维护一个长度为$n$的 $01$序列，每次找到这个序列第$a[i]$个$1$的位置，并把这个$1$变成$0$.

维护这个东西直接树状数组上倍增即可，时间复杂度$O(n\log n)$.

代码如下：

#include<iostream>
#include<cstdio>
#include<algorithm> 
  using namespace std;

#define Abigail inline void
typedef long long LL;

const int N=8000,C=13;

int c[N+9],a[N+9],n,ans[N+9];

void Add(int x,int v){for (;x<=n;x+=x&-x) c[x]+=v;}

int Query_kth(int p){
  int res=0;
  for (int i=C-1;i>=0;--i)
    if (res+(1<<i)<=n&&p>=c[res+(1<<i)]) p-=c[res+=1<<i];
  return res+1;
}

Abigail into(){
  scanf("%d",&n);
  for (int i=2;i<=n;++i)
    scanf("%d",&a[i]);
}

Abigail work(){
  for (int i=1;i<=n;++i) Add(i,1);
  for (int i=n;i>=1;--i){
    ans[i]=Query_kth(a[i]);
    Add(ans[i],-1);
  }
}

Abigail outo(){
  for (int i=1;i<=n;++i)
    printf("%d\n",ans[i]);
}

int main(){
  into();
  work();
  outo();
  return 0;
}