2018.09.24 codeforces 1053C. Putting Boxes Together(线段树)

传送门
就是让你维护动态的区间带权中位数。
然而昨晚比赛时并没有调出来。
想找到带权中位数的中点可以二分(也可以直接在线段树上找)。
也就是二分出第一个断点,使得断点左边的和恰好大于或等于断点右边的和。
现在的问题在于知道断点之后如何统计答案。
我们可以在线段树中维护当前区间全部移到区间最左端点的花费,以及当前区间全部移到区间最右端点的花费。
这样就可以简单合并并轻松统计答案了。
代码:

#include
#define ll long long
#define N 200005
#define mod 1000000007
#define lc (p<<1)
#define rc (p<<1|1)
#define mid (T[p].l+T[p].r>>1)
using namespace std;
inline ll read(){
	ll ans=0,w=1;
	char ch=getchar();
	while(!isdigit(ch)){if(ch=='-')w=-1;ch=getchar();}
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans*w;
}
inline void write(ll x){
	static int buf[66];
	if(!x){putchar('0');return;}
	if(x<0)putchar('-'),x=-x;
	while(x)buf[++buf[0]]=x%10,x/=10;
	while(buf[0])putchar((buf[buf[0]--])^48);
}
struct nd{int l,r,cnt;ll ls,rs,s,sl,sr;}T[N<<2];
int n,q;
ll a[N],w[N];
inline nd operator+(nd a,nd b){return(nd){a.l,b.r,a.cnt+b.cnt,a.ls,b.rs,a.s+b.s,(a.sl+b.sl+(b.rs-b.cnt-a.rs)*(a.s%mod)%mod),(a.sr+b.sr+(b.ls-a.ls-a.cnt)*(b.s%mod)%mod)};}
inline void newnode(int p,int l,int r){T[p]=(nd){l,r,1,a[l],a[l],w[l],0,0};}
inline void build(int p,int l,int r){
	T[p].l=l,T[p].r=r;
	if(l==r)return newnode(p,l,r);
	build(lc,l,mid),build(rc,mid+1,r),T[p]=T[lc]+T[rc];
}
inline void update(int p,int k){
	if(T[p].l==T[p].r)return newnode(p,T[p].l,T[p].r);
	if(k<=mid)update(lc,k);
	else update(rc,k);
	T[p]=T[lc]+T[rc];
}
inline ll query(int p,int ql,int qr){
	if(ql>T[p].r||qr<T[p].l)return 0;
	if(ql<=T[p].l&&T[p].r<=qr)return T[p].s;
	if(qr<=mid)return query(lc,ql,qr);
	if(ql>mid)return query(rc,ql,qr);
	return query(lc,ql,mid)+query(rc,mid+1,qr);
}
inline nd ask(int p,int ql,int qr){
	if(ql<=T[p].l&&T[p].r<=qr)return T[p];
	if(qr<=mid)return ask(lc,ql,qr);
	if(ql>mid)return ask(rc,ql,qr);
	return ask(lc,ql,mid)+ask(rc,mid+1,qr);
}
int main(){
	n=read(),q=read();
	for(int i=1;i<=n;++i)a[i]=read();
	for(int i=1;i<=n;++i)w[i]=read();
	build(1,1,n);
	while(q--){
		int x=read(),y=read();
		if(x<0)w[-x]=y,update(1,-x);
		else{
			int l=x,r=y,ans=l;
			while(l<=r){
				int midd=l+r>>1;
				ll L=query(1,x,midd),R=query(1,midd+1,y);
				if(L>=R)ans=midd,r=midd-1;
				else l=midd+1;
			}
			write((ask(1,x,ans).sl+ask(1,ans,y).sr)%mod),puts("");
		}
	}
	return 0;
}

你可能感兴趣的:(#,线段树,数据结构与分治算法)