poj2182 Lost Cows[BIT二分]

blog题解鸽了许久了。。本来说好的切一题写一个题解的说

---

一个$1 \sim n$数列，已知每个数前面比他小的数个数，试确定该序列。

---

相当的一道水题。可以发现数列最后一个数是首先可以确定下来的。然后把这个数扔掉，看数列倒数第二个数前面有多少更小的其实就是目前剩下的数中的排名。相当于有一颗$n$个点的$1 \sim n$的平衡树，每次查一个排名，找出这个数，并删掉。但是并不想打平衡树。因为这题的操作只涉及简单的删除和查rank操作，可以考虑用BIT代替平衡树。具体是因为BIT本身就是具有树形二分结构的，画图应该可以感觉出来。$C[i]$表示值为i的数有几个（本题中最多$1$个，有些题还要离散化这个$i$），前缀和维护，查rank的时候不断查$C[x]$并二分下去，和平衡树就类似了。有点难讲，不过真的很好理解。

没了。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define dbg(x) cerr<<#x<<" = "<<x<<endl
#define _dbg(x,y) cerr<<#x<<" = "<<x<<"   "<<#y<<" = "<<y<<endl
using namespace std;
template<typename T>inline char MIN(T&A,T B){return A>B?A=B,1:0;}
template<typename T>inline char MAX(T&A,T B){return A<B?A=B,1:0;}
template<typename T>inline T _min(T A,T B){return A<B?A:B;}
template<typename T>inline T _max(T A,T B){return A>B?A:B;}
template<typename T>inline T read(T&x){
    x=0;char c;int f=0;while(!isdigit(c=getchar()))if(c=='-')f=1;
    while(isdigit(c))x=x*10+(c&15),c=getchar();return f?x=-x:x;
}
const int N=100000+7;
int C[N],a[N];
int n;
#define lowbit(x) x&(-x)
inline void Update_add(int x,int k){for(;x<=n;x+=lowbit(x))C[x]+=k;}
inline int Query_sum(int x){int ret=0;for(;x;x-=lowbit(x))ret+=C[x];return ret;}
inline int Query_kth(int x){
    int i=1<<__lg(n);
    while((lowbit(i))^1){
        if(x<=C[i])i-=((lowbit(i))>>1);
        else x-=C[i],i+=((lowbit(i))>>1);
        while(i>n)i-=((lowbit(i))>>1);
    }
    return C[i]^x?i+1:i;
}

int main(){//freopen(".in","r",stdin);freopen(".out","w",stdout);
    read(n);a[1]=1;
    for(register int i=2;i<=n;++i)read(a[i]),++a[i];
    for(register int i=1;i<=n;++i)C[i]=lowbit(i);
    for(register int i=n;i;--i){
        int x=Query_kth(a[i]);
        Update_add(x,-1),a[i]=x;
    }
    for(register int i=1;i<=n;++i)printf("%d\n",a[i]);
    return 0;
}