Lost Cow

LOST COW

问题描述：一共N头奶牛编号从1到N，现在它们的排列顺序发生了错乱，对于每头奶牛，你只知道排在它前面并且编号小于它的奶牛数量。

输入
第1行：单个整数，N ，表示奶牛的总数

之后N-1行：第k行描述了排在第k+1头奶牛之前并且编号小于该奶牛编号的奶牛数量。当然，没有奶牛排在第一头牛的前面，所以没有考虑它。

样例输入
5
1
2
1
0

样例输出
2
4
5
3
1

---

思路：

从后面的奶牛往前数，看后面的奶牛的前面有几头比它编号小的，然后结合储存01的树状数组解决问题。

#include<bits/stdc++.h>
using namespace std;
const int N=8e3+5;
int n,a[N],c[N],b[N],d[N];

int lowbit(int x)
{
    return x&-x;
}

void update(int x,int y)
{
    for(;x<=n;x+=lowbit(x))
        c[x]+=y;
}

int ask(int x)
{
    int temp=0;
    for(;x>0;x-=lowbit(x))
        temp+=c[x];
    return temp;
}

int main()
{
    int l,r,i,j,mid,v,k;
    scanf("%d",&n);
    a[1]=0;
    d[1]=1;
    for(i=1;i<=n;i++)update(i,1);
    for(i=2;i<=n;i++)
    {
        scanf("%d",&a[i]);
        d[i]=i;
    }
    j=1;
    for(i=n;i>0;i--)
    {
        l=0;r=n;
        while(l<r)
        {
            mid=(l+r)/2;
            v=ask(mid);
            if(v<a[i]+1)
                l=mid+1;
            else
                r=mid;
        }
        b[j++]=d[l];
        update(l,-1);
    }
    for(i=j-1;i>0;i--)
        printf("%d\n",b[i]);
    return 0;
}