问题描述
约翰有n头奶牛,这些奶牛长得很相似,约翰经常分不清谁是谁,于是约翰给他们编号1到n,以此来区分每头奶牛。
今天奶牛们想作弄一下约翰,n头奶牛乱序排成一排。
约翰想要分清每个位置对应奶牛的编号。约翰从左起第2头奶牛开始一直到第n头奶牛,依次问每个奶牛一个问题“你左边有多少个编号比你小的奶牛”,奶牛们都会如实回答约翰的问题。
问,你能否根据奶牛的回答,帮助约翰推出这n头牛对应的编号?
输入格式
第一行,一个整数n
接下来n-1行,每行一个整数,分别表示左起第2到第n头奶牛的回答。
输出格式
n行,每行一个整数,对应从左往右每头奶牛的编号
样例输入
5
1
2
1
0
样例输出
2
4
5
3
1
数据范围
1<=N<=50000
树状数组,把数组装好后弄一个二分就行了
为什么会想到二分呢,我也不是很清楚==
看到这种编号自然而然的一种感觉
#include<iostream>
#include<algorithm>
using namespace std;
int n;
int a[50005],ans[50005],c[50005];
int lowbit(int x)
{
return x&-x;
}
int getsum(int x)
{
int sum=0;
for(int i=x;i>0;i-=lowbit(i))sum+=c[i];
return sum;
}
void modify(int x,int y)
{
for(int i=x;i<=n;i+=lowbit(i))c[i]+=y;
}
int main()
{
int mid,Left,Right;
scanf("%d",&n);
for(int i=2;i<=n;i++)scanf("%d",&a[i]);
for(int i=n;i>0;i--) //二分
{
Left=a[i]+1;
Right=a[i]+1+n-i;
while(Left<=Right)
{
mid=(Right+Left)/2;
if(getsum(mid)+a[i]<=mid-1)Right=mid-1;
else Left=mid+1;
}
ans[i]=Left;
modify(Left,1);
}
for(int i=1;i<=n;i++)cout<<ans[i]<<endl;
}