Lost Cows[樹狀陣列+二分]
阿新 • • 發佈:2018-11-23
我們倒過來做,對於當前的x,他前面有dx個比它小,那麼它就是當前的第k大
這個第k大我們二分一下就好了,這個數完了過後,把它在樹狀數組裡的值改為0就可以了
#include<cstdio> #define N 8050 using namespace std; int c[N],n,d[N],a[N]; void Up(int x,int val){for(;x<=n;x+=x&-x)c[x]+=val;} int Q(int x){int ans=0;for(;x;x-=x&-x)ans+=c[x];return ans;} int K(int x){ int l=1,r=n; while(l<r){ int mid=(l+r)>>1; if(Q(mid)>=x) r=mid; else l=mid+1; }return l; } int main(){ scanf("%d",&n); for(int i=2;i<=n;i++) scanf("%d",&d[i]); for(int i=1;i<=n;i++) Up(i,1); for(int i=n;i>=1;i--){ a[i]=K(d[i]+1); Up(a[i],-1); } for(int i=1;i<=n;i++) printf("%d\n",a[i]); return 0; }