Lost Cows poj2182 线段树

    本题若按正序思考，则没什么头绪，但如果我们倒着看会发现，从后向前每一个数字可以找出一头牛。

另外定位一头牛的编号后，为了确定下一头牛，我们需要删除这头牛的编号，即需要区间动态查询，所以可以选择线段树。

思路：可以用每个节点记录该区间中母牛的数目，然后在这棵树中，已知一个节点前面比它小的节点数，我们就可以同过线段树二分快速查找了。

 

#include <iostream>
#include<cstdio>
#include<cstring>

using namespace std;
const int maxn=80005;
int node[maxn*4],pre[maxn];

void build(int l,int r,int rt)
{
    if(l==r)
    {
        node[rt]=1;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    node[rt]=node[rt<<1]+node[rt<<1|1];
}

void update(int l,int r,int x,int rt)
{
    if(l==r&&l==x)
    {
        node[rt]=0;
        return;
    }
    int mid=(l+r)>>1;
    if(x<=mid)
        update(l,mid,x,rt<<1);
    else
        update(mid+1,r,x,rt<<1|1);
    node[rt]--;
}

int query(int l,int r,int c,int rt)
{
    if(l==r)
        return l;
    int mid=(l+r)>>1;
    if(node[rt<<1]>=c)
        return query(l,mid,c,rt<<1);
    else
        return query(mid+1,r,c-node[rt<<1],rt<<1|1);
}

int main()
{
    int i,n;
    while(~scanf("%d",&n))
    {
        for(i=1;i<n;i++)
            scanf("%d",&pre[i]);
        pre[0]=0;
        build(1,n,1);
        for(i=n-1;i>=0;i--)
        {
            pre[i]=query(1,n,pre[i]+1,1);
            update(1,n,pre[i],1);
        }
        for(i=0;i<n;i++)
            printf("%d\n",pre[i]);
    }
}