D. Reverse Sort Sum（思维 + 差分 + 树状数组）

题目链接
太难推了…给我推晕了…

分析

从前往后推:
参考题解

从后往前推：
参考题解1：
参考题解2：

首先：
序列c[]中所有元素之和其实也就是原序列a[]中1的总和 * a[].size()

其次：
就是每个c[i]的贡献分为两部分：
前 i−1 次排序时的值 a[i] 和 后 n−i+1 次排序的值（若干个1，0）;

从后往前推相当于是每次算 后n−i+1次排序 产生的贡献；
每次减掉1排序时产生的贡献，然后判断c[i]是否=i-1，来判断a[i]是否是1

由于我们每次要让一个区间减去一个值 + 单点查询；
所以这个可以用差分 + 树状数组维护。

具体看代码！

代码

#include<iostream>
#include<queue>
#include<cstring>
#include<vector>
#include<stdio.h>
#include<map>
#include<algorithm>
#include<deque>
#include<stack>
#include<set>
// #include <unordered_map>
#include<math.h>
#include<string.h>
#define IOS ios::sync_with_stdio(false),cin.tie(0);
using namespace std;
 
#define pb push_back
#define coutl cout<<"------------"<<endl;
#define fi first
#define se second

#define ire(x) scanf("%d",&x)
#define iire(a,b) scanf("%d %d",&a,&b)
#define lre(x) scanf("%lld",&x)
#define llre(a,b) scanf("%lld %lld",&a,&b)
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define endl "\n"
#define PI acos(-1.0)
//#define int long long
typedef long long ll;
typedef unsigned long long ull;
      
typedef pair<int, int> PII;
typedef pair<double, int> PDI;
typedef pair<ll, ll> PLL;
typedef pair<double, double> PDD;
typedef pair<double, pair<int, double> > PDID;
typedef pair<char, char> PCC;
typedef pair<char, pair<int, int> > PCII;
typedef pair<int, pair<int, int> > PIII;
typedef pair<int, pair<int, pair<int, int> > > PIIII;
typedef pair<ll, pair<int, int> > PLII;

const int maxn = 2e5 + 7;
const int N = 1e6 + 7;
const int M = 1e6 + 7;
const int mod = 3*5*7*11*13*17*19*23;
const int inv = mod - mod/2;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const double pi = acos(-1);
const double eps = 1e-8;
  
ll gcd(ll a,ll b) {return b==0 ? a : gcd(b,a%b);}
ll lcm(ll a,ll b) {return a*b / gcd(a,b);}
ll qmi(ll a,ll b,ll p) {ll ans = 1; while(b) { if(b & 1) ans = ans * a % p; a = a * a % p; b >>= 1; } return ans;}
int lowbit(int x) {return x & (-x);}

int c[maxn];
int a[maxn];
int n;
int tr[maxn];	//树状数组维护差分值

void update(int x,int v)
{
	while(x <= n)
	{
		tr[x] += v;
		x += lowbit(x);
	}
}

int query(int x)
{
	int ans = 0;
	while(x)
	{
		ans += tr[x];
		x -= lowbit(x);
	}
	return ans;
}

void solve()
{
	ire(n);
	for(int i=1;i<=n;i++) tr[i] = a[i] = 0;
	
	ll sum = 0;
	for(int i=1;i<=n;i++)
	{
		ire(c[i]);
		update(i,c[i]-c[i-1]);
		
		sum += c[i];
	}
	
	sum /= n;
	for(int i=n;i>=1;i--)
	{
		if(sum)
		{
			update(i-sum+1, -1);
			
			int num = query(i);
			if(num == i-1) a[i] = 1, sum--;
		}
		else break;
	}
	
	for(int i=1;i<=n;i++) cout<<a[i]<<' ';
	cout<<'\n';
}

int main()
{
	int t;
	ire(t);
	while(t--)
	{
		solve();
	}
	
	return 0;
}