题目链接
太难推了…给我推晕了…
分析
从前往后推:
参考题解
首先:
序列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;
}