谜一样的牛(AcWing-244)

写在前面：参考大佬博客Initialize.

 

题目大意：

有 nn 头奶牛，已知它们的身高为 1∼n 且各不相同，但不知道每头奶牛的具体身高。

现在这 nn 头奶牛站成一列，已知第 i头牛前面有 AiAi 头牛比它低，求每头奶牛的身高。

输入格式

第 11 行：输入整数 n。

第 2..n2..n 行：每行输入一个整数 Ai，第 i行表示第 i 头牛前面有Ai 头牛比它低。
（注意：因为第 1 头牛前面没有牛，所以并没有将它列出）

输出格式

输出包含 n 行，每行输出一个整数表示牛的身高。

第 i行输出第 i 头牛的身高。

思路分析：

啊不会,QAQ,完全没思路。弱弱的我WWW

-------------------------------------------------------------------------------------------分割线

好我们好好分析下：如果我们考虑1-n中间的点 我们只知道他前面有多少比他小，它后面的情况我们不知道，有好多情况。 如果一些点确定了那？ 还真有：我们注意点n，n后面没有点 而且每一头牛都是1-n的高度，而且唯一！ 这就是我们想要的！！ 编号为n的点的答案必是n+1  确定了一个点,我们倒着去找下个点(这样只需要考虑这个点的左边)  在余下的点里找到符合身高要求的点。

整理一下： 我们倒着去找答案 每次在余下未选定的点中找到符合要求的身高。

代码实现思路：

用01串来记录一个点（答案）是否已被选 0已被选 1未选 借用树状数组来更新一个点的前缀和(也就是在当前情况下比这个点小的个数（包含这个点）)。在查找答案时可以用二分。

 AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int n,a[N],ans[N];  //ans来记录答案
int c[N];
int lowbit(int x){
	return x&(-x);
} 
void add(int x,int num){  //更新
	for( ;x<=n;c[x]+=num,x+=lowbit(x));
}
int query(int x){  //查找一个点的前缀和
	int re=0;
	for( ;x>0;re+=c[x],x-=lowbit(x));
	return re;
}
void change(){   //初始建立树状数组
	for(int i=1;i<=n;i++){
		add(i,1);
	}
}
int getans(int k){  
	int l=1,r=n;
	while(l<r){   //二分符合条件的答案
		int mid=l+r>>1;
		if(query(mid)>=k) r=mid;
		else l=mid+1;
	}
	add(l,-1);
	return l;
}
int main(){
	scanf("%d",&n);
	change();
	for(int i=2;i<=n;i++){
		scanf("%d",&a[i]);
	}
    // n 的答案已确定
	ans[n]=a[n]+1,add(a[n]+1,-1);  //更新树状数组
	for(int i=n-1;i>=1;i--){
		ans[i]=getans(a[i]+1);  //记录答案
	}
	for(int i=1;i<=n;i++){
		printf("%d\n",ans[i]);
	}
	return 0;
}