题目大意:一群牛的编号为1~n,现在乱站成一列,fj记录下来了每头牛前面编号比它编号小的牛的头数,求各个位置上牛的编号
解题思路:从后往前推,假如排在最后的一头牛比他编号小的数量为a,那么它的编号必然为a+1。我们把编号为a+1的这头牛删掉,假如排在倒数第二的一头牛比他编号小的数量为b,那么该牛就为删掉编号后剩余牛中的第b+1头牛,我们可以照此思路下去...
问题就可以转化为搜索排在第k位的数为多少,我们可以用线段树来实现。
#include<iostream>
#define maxn 10010
using namespace std;
int n;
int a[maxn];
int re[maxn];
struct stu
{
int l,r,sum;
};
stu mapp[maxn*4];
void input()
{
a[0]=0;
for(int i=1;i<n;i++) cin>>a[i];
}
void build(int l,int r,int count)
{
mapp[count].l=l;
mapp[count].r=r;
mapp[count].sum=r-l+1;
if(l==r) return;
int mid=(l+r)/2;
build(l,mid,count*2);
build(mid+1,r,count*2+1);
}
int find(int k,int count)
{
mapp[count].sum--;
if(mapp[count].l==mapp[count].r) return mapp[count].l;
if(mapp[count*2].sum>=k) return find(k,count*2);
else return find(k-mapp[count*2].sum,count*2+1);
}
void solve()
{
for(int i=n-1;i>=0;i--)
{
int x=find(a[i]+1,1);
re[i]=x;
}
}
void output()
{
for(int i=0;i<n;i++) cout<<re[i]<<endl;
}
int main()
{
cin.sync_with_stdio(false);
while(cin>>n)
{
input();
build(1,n,1);
solve();
output();
}
return 0;
}