D. Reverse Sort Sum

我们首先能反应出来的是可以倒着看 并且第一个样例来看好像最后一定是1
这是因为我们最后一个样例的下标和自身的数已经相等了那么我们可以等到一定是1
要是他是1的话 我们考虑可不可以把他砍掉 我们砍掉的话 是砍掉最后一步最后一步的数肯定就是最后那几位减一
还有我们可以通过总和等到数有几个1 这个很好想到
那么对于每一位 我们从后面开始看 要是这一位自身的数==pos下标 那么我们向前长度为one的区间-1 然后再往前找即可
因为这里我们一定可以找到那么几个1 所以说是不会越界的

#include <bits/stdc++.h>
using namespace std;
const int N = 2e5+10;
const int M = 998244353;
const int mod = 1000000007;
#define int long long
#define endl '\n'
#define all(x) (x).begin(),(x).end()
#define YES cout<<"YES"<<endl;
#define NO cout<<"NO"<<endl;
#define _ 0
#define INF 0x3f3f3f3f3f3f3f3f
#define fast ios::sync_with_stdio(false);cin.tie(nullptr);
int tr[N],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;
}
int query(int x){
    int res=0;
    for(int i=x;i;i-=lowbit(i))res+=tr[i];
    return res;
}
void upd(int l,int r,int x){
    add(l,x),add(r+1,-x);
}
void solve() {
    cin>>n;
    for(int i=0;i<=n;i++)tr[i]=0;
    int sum=0;
    vector<int>a(n+1);
    for(int i=1;i<=n;i++){
        cin>>a[i];sum+=a[i];
        add(i,a[i]),add(i+1,-a[i]);
    }
    int one=sum/n;
    vector<int>v(n+1);
    for(int i=n;i>=1;i--){
        if(query(i)>=i){
            v[i]=1;
            add(i-one+1,-1),add(i+1,1);
            one--;
        }
    }
    for(int i=1;i<=n;i++)cout<<v[i]<<' ';cout<<endl;
}
signed main(){
    fast
    int T;cin>>T;
    while(T--) {
        solve();
    }
    return ~~(0^_^0);
}

标签：const,782,int,Codeforces,我们,--,add,Round,define
来源： https://www.cnblogs.com/ycllz/p/16689228.html