谜一样的牛

谜一样的牛

问题描述

---

题意解释

从题目在中可以知道，这n头牛排列在一起后，是按从矮到高排序的，身高是逐渐递增的。题目输入n=5，说明这5头牛的身高范围给定了，是在1~5之间。然后输入1 2 1 0，没有输入第一头牛前面有多少头牛比它矮，但是因为它是第一头牛，它前面不可能有牛，所以暗含第一头牛前面有0头牛比它矮。对于第二头牛来说，表示它前面有1头牛比它矮；对于第三头牛来说，它前面有2头牛比它矮；对于第四头牛来说，它前面有1头牛比它矮；对于第五头牛来说，它前面有0头牛比它矮。对于样例来说，不可能从前往后求第i头牛的身高，但是从后往前看，就可以倒推出来牛的身高了。A[5] = 0，含义为 前五个数中，没有（0个）比它小的数了，那么它的身高排位一定是1(表示第1矮);还剩下可选的身高排位有2 3 4 5；A[4] = 1，含义为 剩下的四个数{2,3,4,5}中，有一个比它小的数，那么它是第二小，所以它的身高排位一定是3(表示第3矮)，还剩下可选的身高排位有{2,4,5}；A[3] = 2，含义为 剩下三个数{2,4,5}中，有两个比它小的数，那么它是第三小，所以它的身高排位一定是5(表示第5矮)，还剩下可选的身高排位有{2,4}；A[2] = 1，含义为剩下两个数{中，它是第二小的，所以它的身高排位一定是4{表示第4矮}，还剩下可选的身高排位有{2}；A[1]=0，含义是有0头牛比它矮，因为此时剩余可选的身高排位只剩下{2}了，因此它的身高排位一定是2(表示第2矮)。因此，第一头牛到第五头牛的身高排位如下：

• 第一头牛，身高排位2，表示第2矮
• 第二头牛，身高排位4，表示第4矮
• 第三头牛，身高排位5，表示第5矮
• 第四头牛，身高排位3，表示第3矮
• 第五头牛，身高排位1，表示第1矮

---

核心思路

如果在最后一头牛前面有$A_n$有牛比它矮，那么它的身高排位一定是$H_n=A_n+1$。如果倒数第二头牛前面有$A_{n-1}$有牛比它矮，那么：

• 若$A_{n-1}<A_n$，则它的身高排位是$H_{n-1}=A_{n-1}+1$
• 若$A_{n-1}\ge A_n$，则它的身高排位是$H_{n-1}=A_{n-1}+2$

以此类推，如果第$k$头牛前面有$A_k$头牛比它矮，那么它的身高排位$H_k$就从可选身高1~n中选出第$A_k+1$小的那个身高排位，并且这个身高排位它在$H_{k+1},H_{k+2},\cdots ,H_n$中没有出现过。

建立一个长度为n的01序列exist，exist[i]为1表示身高排位i还没有被选走，为0表示已经被选择了，那就要在剩余的可选身高排位中，第k小的身高排位，也就是要找到最小的x，让exist序列以x位置结尾的前缀和刚好等于k，即$sum(x)=k$。exist序列由于在选择了牛的高度后动态变化，又要维护其前缀和，刚好符合树状数组的特性，用树状数组维护exist序列的前缀和即可，每确定一头牛的高度，就从exist序列中把对对应位置设为0即可。

树状数组中query(x)，求的是exist[1]~exist[x]的前缀和，有因为exist序列中只有1或0，所以所以query(x)的值为多少就代表着exist[1]到exist[x]中还有多少个数为1（这个身高还没被选中）。因为这n头牛的身高排位是在1到n之间，且是单调递增的。

如何理解sum(x)=k呢？

首先，$sum(x)$表示区间$[1,x]$内exist序列的前缀和，可以知道此时exist序列中还有多少个1，但是我们只是知道有多少个1，但是并不知道这些1的位置具体分布。不妨设k=3，我们要在剩余可选的身高排位中找出第k=3矮的身高。因为通过$sum(x)$我们只知道exist序列中还有多少个1，因此此时$exist$序列共有10种可能，如下图分析。

1. 对于第一种序列，那么要想使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
2. 对于第二种序列，那么要想使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
3. 对于第三种序列，那么想要使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
4. 对于第四种序列，那么想要使得$sum(x)=k=3$，那么此时$x=4$，也就是说只可以用身高排位4。
5. 对于第五种序列，那么想要使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
6. 对于第六种序列，那么想要使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
7. 对于第七种序列，那么想要使得$sum(x)=k=3$，那么此时$x=4$，也就是说只可以用身高排位4。
8. 对于第八种序列，那么想要使得$sum(x)=k=3$，那么此时$x=5$，也就是说只可以用身高排位5。
9. 对于第九种序列，那么想要使得$sum(x)=k=3$，那么此时$x=4$，也就是说只可以用身高排位4。
10. 对于第十种序列，那么想要使得$sum(x)=k=3$，那么此时$x=3$，也就是说只可以用 身高排位3。

由此可知,对于第十种序列，$x=3$，是最小的能够使得$sum(x)=k$的身高排位，那么此时就可以选择身高排位3，而不是选择身高排位4和身高排位5。也就是说我们想寻找第k小的数，那么可以转换为求使得$sum(x)=k$的情况下$x$取得最小，$x=4$是身高排位4，第4矮；$x=5$是身高排位5，第5矮；而$x=3$是身高排位3，第3矮。由于第3矮<第4矮<第5矮，因此应该选择$x=3$。然而从1开始一个一个向后query找x太慢了，发现从1 到 n的query(i) 满足单调且连续的性质，所以这道题目可以用二分求解。由之前的图可知，$sum(x)$是单调递增的，所以可以用二分。

方法：树状数组+二分，从后往前枚举时，有两个操作：

• 从剩余的数中找第k小的数
• 删除某个数

如何理解用树状数组来维护序列exist数组呢？

我们知道树状数组由两个经典函数，一个是单点修改update函数，另一个是区间查询getSum函数。

• 对于单点修改update(int x,int v)函数，表示在x位置+v，那么如何知道这个x值和v值呢？其实这个x值也就是exist中的i，这个v值也就是exist序列中的某个数exist[i]，因此，树状数组的update函数中的x值和v值来源于exist序列，作用于exist序列上，所以称用树状数组来维护序列exist数组。
• 对于区间查询getSum(int x)函数，如何知道这个x值呢？其实这个x值也就是exist中的i，因此，树状数组的getSum函数中的x值来源于exist序列，作用于exist序列上，所以称用树状数组来维护序列exist数组。

---

#include<iostream>
using namespace std;
const int N=100010;
int n;
int h[N];	//记录第i头牛前面由Ai头牛比它矮
int ans[N];	//最终这1~n头牛的身高排位
int C[N];	//树状数组
int lowbit(int x)
{
    return x&(-x);
}
//单点修改
void update(int x,int v)
{
    for(int i=x;i<=n;i+=lowbit(i))
    C[i]+=v;
}
//区间查询，返回区间[1,x]的前缀和
int getSum(int x)
{
    int sum=0;
    for(int i=x;i>0;i-=lowbit(i))
    sum+=C[i];
    return sum;
}
int main()
{
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    scanf("%d",&h[i]);
    //树状数组初始化       序列exist全为1
    for(int i=1;i<=n;i++)
    update(i,1);
    //从第n头牛倒序逆推到第1头牛
    for(int i=n;i>0;i--)
    {
        int k=h[i]+1;	//第k小
        int l=1,r=n;
        //二分
        while(l<r)
        {
            int mid=l+r>>1;
            if(getSum(mid)>=k)
            r=mid;
            else
            l=mid+1;
        }
        //此时找到了最小的位置x,它就是r或者l
        ans[i]=r;	//这里可以写出ans[i]=l
        update(r,-1);	//这里可以写成update(l,-1);
    }
    for(int i=1;i<=n;i++)
    printf("%d\n",ans[i]);
    return 0;
}

---