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