雅礼集训2019 D7T2 Subsequence

雅礼集训2019 D7T2 Subsequence

直接贴题解：

平衡树代码：

#include<bits/stdc++.h>
#define ll long long
#define N 100005

using namespace std;
inline int Get() {int x=0,f=1;char ch=getchar();while(ch<'0'||ch>'9') {if(ch=='-') f=-1;ch=getchar();}while('0'<=ch&&ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}return x*f;}

int n;
ll a[N];
int ch[N][2],fa[N];
int size[N];
ll p[N];
#define ls ch[v][0]
#define rs ch[v][1]
int rt;
void update(int v) {size[v]=size[ls]+size[rs]+1;}
ll add[N];

void Add(int v,ll f) {
	p[v]+=f;
	add[v]+=f;
}

void down(int v) {
	if(add[v]) {
		if(ls) Add(ls,add[v]);
		if(rs) Add(rs,add[v]);
		add[v]=0;
	}
}

void rot(int v) {
	int f=fa[v],gr=fa[f];
	int sn=v==ch[f][1],son=ch[v][!sn];
	if(gr) ch[gr][f==ch[gr][1]]=v;
	ch[f][sn]=son;
	ch[v][!sn]=f;
	fa[v]=gr,fa[f]=v;
	if(son) fa[son]=f;
	update(f),update(v);
}

void splay(int v) {
	static int st[N],top;
	st[top=1]=v;
	for(int i=v;fa[i];i=fa[i]) st[++top]=fa[i];
	for(int i=top;i>=1;i--) down(st[i]);
	while(fa[v]) {
		int f=fa[v],gr=fa[f];
		if(gr) rot(f==ch[gr][1]^v==ch[f][1]?v:f);
		rot(v);
	}
	rt=v;
}

int tot;
int Find(int k) {
	int v=rt;
	while(1) {
		if(size[ls]+1==k) return v;
		if(size[ls]>=k) v=ls;
		else k-=size[ls]+1,v=rs;
	}
}

int Find_rk(int v) {
	splay(v);
	return size[ls]+1;
}

void Insert(int &v,int k,int id) {
	if(!v) {
		v=id;
		update(v);
		return ;
	}
	if(size[ls]>=k-1) {
		Insert(ls,k,id);
		fa[ls]=v;
	} else {
		Insert(rs,k-size[ls]-1,id);
		fa[rs]=v;
	}
	update(v);
}

int pos;
void binary(int v,ll a,int pre,int &pos) {
	if(!v) return ;
	down(v);
	ll now=pre+size[ls]+1;
	if(a*now>=p[v]) {
		pos=v;
		binary(ls,a,pre,pos);
	} else binary(rs,a,pre+size[ls]+1,pos);
}

void out(int v,ll &ans) {
	if(!v) return ;
	down(v);
	out(ls,ans);
	ans+=p[v];
	cout<<ans<<" ";
	out(rs,ans);
}

int main() {
	n=Get();
	for(int i=1;i<=n;i++) a[i]=Get();
	rt=1;
	update(rt);
	p[rt]=a[1];

	for(int i=2;i<=n;i++) {
		binary(rt,a[i],0,pos=i);
		if(pos!=i) splay(pos);
		int rk=pos==i?i:Find_rk(pos);
		p[i]=a[i]*rk;
		Insert(rt,rk,i);
		splay(i);
		if(ch[i][1]) Add(ch[i][1],a[i]);
	}
	ll ans=0;
	out(rt,ans);
	return 0;
}