N头奶牛排队,它们的身高为1~n,知道每头牛前面有多少头比自己矮,求每头牛的身高。
第一头牛前面是0,我们不知道他多高,但最后一头牛前面有n个比自己矮的话就可以知道他的高度是n+1
如果知道最后一头牛的高度是n+1,然后知道倒数第二头牛前面有m个比自己矮,那么就知道了倒数第二头牛的高度为m+1(m>=n),或者m(m<n)
其实就是动态的维护前缀和,并且单点查询
#include <bits/stdc++.h> #define form(i,n) for(int i=1;i<=n;i++) #define forn(i,n) for(int i=0;i<n;i++) #define mst(a) memset(a,0,sizeof(a)) using namespace std; typedef pair<int,int> P ; const int N=1e5+90; typedef unsigned long long ull; typedef long long ll; const int MAX_N=4e5+20; ll a[N],b[N],c[N],n; ll lowbit(ll x){ return x&-x; } void add(ll x,ll y){ while (x<=n){ b[x]+=y; x+=lowbit(x); } } ll ask(ll x){ ll ans=0; while (x){ ans+=b[x]; x-=lowbit(x); } return ans; } int main(){ cin>>n; for(int i=1;i<=n;i++)b[i]=lowbit(i); for(int i=2;i<=n;i++){ scanf("%d",&c[i]); } for(int i=n;i;i--){ int k=c[i]+1; ll l=1,r=n; while (l<r){ ll mid=l+r>>1; if(ask(mid)>=k) r=mid; else l=mid+1; } a[i]=l; add(l,-1); } for(int i=1;i<=n;i++)printf("%d\n",a[i]); }