题意:给2组数据a和b数组,每次有2种操作:(+,l,r,x)把a数组第l个到第r个元素全置为x,(?,l,r)查询[l,r]之间哪些位置满足a[i]>=b[i](i>=l && i<=r)并把这些位置的数量统计
一直想很久,没想到什么有效的方案,直到看到题解才明白过来,原来线段树套平衡树还有这种情况:里面其实不是平衡树,只是有序表。
然后这题就转换为区间查找数对应排名
由于此题不用对2个数组都修改,其中1个b树可作为固定的线段树套有序表以节省时间,另外1个表a树则单纯使用线段树的方法先修改,再更新对应b树结点的排名
这里查找排名如果全部logn查找会因为常数太大直接卡,注意每个结点都含有序表并且上下有包含关系
那咱们可以在b树自底向上更新父结点排名对应左右子树里的排名,用归并排序的方法,占用空间才o(nlogn),时间也是o(nlogn)
顺带把会改变的a树1个个结点查询b树查出排名,修改时先查出根结点对应位置,再根据位置子树表一边向下更新一边转移到子树对应位置
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#include<stack>
#include<math.h>
#include<vector>
#include<map>
#include<set>
#include<stdlib.h>
#include<cmath>
#include<string>
#include<algorithm>
#include<iostream>
using namespace std;
typedef __int64 ll;
int max(int a,int b){return a>b?a:b;}
int min(int a,int b){return a<b?a:b;}
int cnt;
const int N=,M=,E=;
int n,m,i,a[N],b[N],x,l,r;
int st[M],en[M],v[M],tag[M],pl[E],pr[E],pool[E],cur;
ll ans,sum; void build(int x,int l,int r)
{
tag[x]=-;
if(l==r)
{
st[x]=cur+;
pool[++cur]=b[l];
en[x]=cur;
v[x]=(a[l]>=b[l]);
return;
}
int mid=((l+r)>>);
build(x<<,l,mid);
build((x<<)|,mid+,r);
v[x]=v[x<<]+v[(x<<)|];
int al=st[x<<],ar=en[x<<],bl=st[(x<<)|],br=en[(x<<)|];
st[x]=cur+;
while(al<=ar&&bl<=br)pool[++cur]=pool[al]<pool[bl]?pool[al++]:pool[bl++];
while(al<=ar)pool[++cur]=pool[al++];
while(bl<=br)pool[++cur]=pool[bl++];
en[x]=cur;
al=st[x<<],bl=st[x<<|];
for(int i=st[x];i<=cur;i++)
{
while(al<=ar&&pool[al]<=pool[i])al++;
while(bl<=br&&pool[bl]<=pool[i])bl++;
pl[i]=al-,pr[i]=bl-;
if(pl[i]<st[x<<])pl[i]=;
if(pr[i]<st[(x<<)|])pr[i]=;
}
} inline void rankpt(int x,int p)
{
v[x]=(p?p-st[x]+:);
tag[x]=p;
}
inline void pushdown(int x)
{
if(tag[x]<)return;
int p=tag[x];
rankpt(x<<,pl[p]);
rankpt((x<<)|,pr[p]);
tag[x]=-;
} void update(int x,int a,int b,int p)
{
if(l<=a && b<=r){rankpt(x,p);return;}
pushdown(x);
int mid=(a+b)>>;
if(l<=mid)update(x<<,a,mid,pl[p]);
if(r>mid)update((x<<)|,mid+,b,pr[p]);
v[x]=v[x<<]+v[(x<<)|];
} void query(int x,int a,int b)
{
if(l<=a && b<=r)
{
ans+=v[x];
return;
}
pushdown(x);
int mid=((a+b)>>);
if(l<=mid)query(x<<,a,mid);
if(r>mid)query((x<<)|,mid+,b);
v[x]=v[x<<]+v[(x<<)|];
} inline int lower(int x){
//lower_bound(pool+st[1],pool+ed[1]+1,x);
int l=st[],r=en[],mid,t=;
while(l<=r)
if(pool[mid=(l+r)>>]<=x)l=(t=mid)+;
else r=mid-;
return t;
} int seeda, seedb, C = ~(<<), MM = (<<)-;
int rnd(int last) {
seeda = ( + (last >> )) * (seeda & MM) + (seeda >> );
seedb = ( + (last >> )) * (seedb & MM) + (seedb >> );
return (C & ((seeda << ) + seedb)) % ;
} int main()
{
int t,ku;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d%d",&n,&m,&seeda,&seedb);
for(i=;i<=n;i++)scanf("%d",a+i);
for(i=;i<=n;i++)scanf("%d",b+i);
ans=sum=cur=;
build(,,n);
for(i=;i<=m;i++)
{
l=rnd(ans)%n+,r=rnd(ans)%n+,ku=rnd(ans)+;
int kkk=lower(ku);
if(l>r)l^=r^=l^=r;
if((l+r+ku)&)update(,,n,lower(ku));
else
{
ans=;
query(,,n);
sum=(sum+(ll)i*ans)%;
}
}
printf("%I64d\n",sum);
}
return ;
}