D. Restore Permutation

D. Restore Permutation

就是给一个n个数的全排，然后bi记录比ai小且在排在ai前面的数的和,求ai

树状数组维护，二分

#include<bits/stdc++.h>
using namespace std;
#define int long long
#define si signed
#define endl '\n'
#define sc(x) scanf("%I64d",&x);
#define read(A) for(int i=1;i<=n;i++)scanf("%I64d",&A[i]);
#define P pair<int,int>
#define fi first
#define se second
#define ot(x) cout<<x<<'\n';
#define maxn 200000+500
int A[maxn];
int n,t,x,y,a,b;
int c[maxn];
int ans[maxn];
bool B[maxn];
int lowbit(int x)
{
    return x&(-x);
}

void add(int i,int k)
{
    while(i <= n+)
    {
        c[i] += k;
        i += lowbit(i);
    }
}

int get(int i)
{

    int res = ;
    while(i > )
    {
        res += c[i];
        i -= lowbit(i);
    }
    return res;
}
int solve(int x)
{

    int l=,r=n;
    int mid;
    while(l+<r)
    {
        mid=(l+r)/;
        if(get(mid-)<=x){
            l=mid;
        }else if(get(mid-)>x){
            r=mid-;
        }
    }
    int ans;
    if(get(r-)==x){
        ans=r;
    }else ans=l;

    return ans;
}
signed main()
{
    sc(n);

    for(int i=;i<=n;i++){
       sc(A[i]);
       add(i,i);
    }
    //
    int _j=;

    for(int i=n; i>; i--)
    {
        if(A[i]==)
        {
            for(_j; _j<=n; _j++)
            {
                if(!B[_j])
                {
                    ans[i]=_j;
                    add(_j,-_j);
                    B[_j]=;
                    break;
                }
            }
        }
        else
        {
            ans[i]=solve(A[i]);
            B[ans[i]]=;
            // cout<<ans[i]<<'\n';
            add(ans[i],-ans[i]);
        }
    }
    for(int i=; i<=n; i++)
    {
        cout<<ans[i]<<' ';
    }

}