CF1659D Reverse Sort Sum题解

题意 ：某个长度为n01序列，每次sort只sort前1-n个数，然后把这n次sort结果加起来的数列给你，反推这个序列

从n反推到1，如果此位置为i，则一定为1，反之一定为0，可以使用差分树状数组撤销掉之前n的影响，可以直接递推下来

#include <bits/stdc++.h>
using namespace std;
#define int long long
int n,ans[201000],t[201000],a[200100];
void add(int x,int val){
	while(x<=n){
		t[x]+=val;
		x+=x&-x;
	}
}
int sum(int x){
	int anss=0;
	while(x){
		anss+=t[x]; 
		x-=x&-x;
	}return anss;
}
signed main()
{
int tt;
cin>>tt;
while(tt--){
	cin>>n;int summ=0;
	for(int i=1;i<=n;i++)ans[i]=t[i]=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];summ+=a[i];
		add(i,a[i]);add(i+1,-a[i]);
	}summ/=n;
	for(int i=n;i>=1;i--){
		if(sum(i)>=i){
			ans[i]=1;
			add(i-summ+1,-1);
			add(i+1,-1);
			summ--;
		}
	}
	for(int i=1;i<=n;i++)printf("%lld ",ans[i]);
	printf("\n");
}	
}