D. Reverse Sort Sum

题目D. Reverse Sort Sum

考点： 线段树

题意：有一个长的为 $n$ ，由 $0$ 和 $1$ 组成的数组，定义一个函数 $f(k,A)$ ,它返回另一个数组 $B$ ，即对 $A$ 的前 $k$ 个元素按非递减的顺序排序的结果。例如，$f(4,[0,1,1,0,1,0])=[0,0,1,1,1,0]$ 。注意，前 $4$ 个元素被排序。现考虑由 $f(1,A),f(2,A),\cdot \cdot \cdot ,f(n,A)$ 生成数组 $B_1,B_2,\cdot \cdot \cdot ,B_n$ 。$C$ 是通过取 $B_1,B_2,\cdot \cdot \cdot ,B_n$ 的元素之和得到的数组。

例如，让 $A=[0,1,0,1]$ ，则有 $B_1=[0,1,0,1],B_2=[0,1,0,1],B_3=[0,0,1,1],B_4=[0,0,1,1]$ ，得到的 $C=B_1+B_2+B_3+B_4=[0,2,2,4]$ 。

现在得到数组 $C$ ，需要按照上述的方法确定数组 $A$ 。

思路：

• 利用公式 $k=\frac{{\sum }_1^n{C}_i}{n}$ 可以求出数组 $A$ 中有都少个 $1$ ，利用最后一行的有序性，最后 $k$ 个值一定为 $1$ 。
• $B_1$ 只有第 $1$ 个有序，$B_2$ 只有前 $2$ 个有序，$B_3$ 只有前 $3$ 个有序，依此类推，$B_{n-1}$ 只有前 $n-1$ 个有序。当确定的第 $n$ 行的第 $n$ 个值时，用 $\frac{C_n-k_n}{n-1}$ 便是 $A_n$ 的值。
• 在求 $A_{n-1}$ 的时间需要先将 $C_{k,...,n}-1$ ，再用公式 $k=\frac{{\sum }_1^{n-1}{C}_i}{n-1}$ ，接下来的思路与第二点相同。

需要注意的是，本题的数据会爆 $int$ ，所以需要用 $longlong$

#include<stdio.h>

typedef long long ll;
const int N = 2e5 + 10;
ll a[N],c[N];
struct node{
	int l,r;
	ll b,x;
}tr[4 * N];

void up(int p) {
	if(tr[p].x == 0) return ;
	tr[p * 2].b += tr[p].x * (tr[p * 2].r - tr[p * 2].l + 1);
	tr[p * 2 + 1].b += tr[p].x * (tr[p * 2 + 1].r - tr[p * 2 + 1].l + 1);
	
	tr[p * 2].x += tr[p].x;
	tr[p * 2 + 1].x += tr[p].x;
	tr[p].x = 0;
}

void build(int p,int l,int r) {
	tr[p].l = l;tr[p].r = r;tr[p].x = 0;tr[p].b = 0;
	if(l == r) {
		scanf("%lld",&tr[p].b);
		return ;
	}
	
	int mid = (l + r) >> 1;
	build(p*2,l,mid);
	build(p*2+1,mid+1,r);
	
	tr[p].b = tr[p*2].b + tr[p*2+1].b;
}

void modify(int p,int l,int r,int x) {
	if(l <= tr[p].l && r >= tr[p].r) {
		tr[p].b += x * (tr[p].r - tr[p].l + 1);
		tr[p].x += x;
		return ;
	}
	
	up(p);
	
	int mid = (tr[p].l + tr[p].r) >> 1;
	
	if(mid >= l) modify(p*2,l,r,x);
	if(mid < r) modify(p*2+1,l,r,x);
	
	tr[p].b = tr[p*2].b + tr[p*2+1].b;
}

ll query(int p,int l,int r) {
	if(tr[p].l >= l && r >= tr[p].r) return tr[p].b;
	
	up(p);
	
	int mid = (tr[p].l + tr[p].r) >> 1;
	
	ll k = 0;
	
	if(mid >= l) k += query(p*2,l,r);
	if(mid < r) k += query(p*2+1,l,r);
	
	return k;
}

ll ask(int p,int k) {
	if(tr[p].l == tr[p].r) return tr[p].b;
	
	up(p);
	
	int mid = (tr[p].l + tr[p].r) >> 1;
	
	if(mid >= k) return ask(p*2,k);
	else return ask(p*2+1,k);
}

int main(){
	int t;
	scanf("%d",&t);
	while(t --) {
		int n;
		scanf("%d",&n);
		build(1,1,n);
		
		for(int i=n;i>=1;i--) {
			ll L = query(1,1,i) / i;
			
			if(i == 1) {
				a[i] = L;
				break;
			}
			
			if(L > 0) modify(1,i-L+1,i,-1);
			
			ll b = ask(1,i);
			
			a[i] = b / (i - 1);
			
		}
		for(int i=1;i<=n;i++) printf("%lld ",a[i]);
		printf("\n");
	}
}