莫欺少年穷,修魔之旅在这开始—>算法提高课题解
思路:
1. sum(x) = k 表示 x 就是在剩下的数中的第 k 小数
2. 用 1 表示该数字还未用过,如果用过就删除该数字
3. 从后往前遍历,用二分找出 sum(x) = k 的数,并删去它
#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
int n;
int h[N];
int tr[N];
int ans[N];
int lowbit(int x)
{
return x&-x;
}
void add(int x,int c)
{
for(int i=x;i<=n;i+=lowbit(i)) tr[i]+=c;
}
// sum(x) = k 表示 x 就是在剩下的数中的第 k 小数
int sum(int x)
{
int res=0;
for(int i=x;i;i-=lowbit(i)) res+=tr[i];
return res;
}
int main()
{
cin>>n;
for(int i=2;i<=n;i++) cin>>h[i];
//用 1 表示这个数还没用过
for(int i=1;i<=n;i++) add(i,1);
for(int i=n;i;i--)
{
int k=h[i]+1;
int l=1,r=n;
while(l<r)
{
int mid=l+r>>1;
if(sum(mid)>=k) r=mid;
else l=mid+1;
}
ans[i]=l;
//把这个数删掉
add(l,-1);
}
for(int i=1;i<=n;i++) cout<<ans[i]<<endl;
return 0;
}