Hdu5737-Differencia(有序表线段树)

时间:2022-05-15 16:33:17

题意很直观,我就不说了。

解析:这是我以前没有接触过的线段树类型,有序表线段树,每个节点申请了两段空间,主要是为了保存左边儿子会有多少比v小的,右边儿子会有多少比v小

的,所以在建树过程中要归并排序。可能我讲起来比较难懂,详见代码,我给了注释。

代码

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
typedef __int64 LL;
#define e tree[id]
#define lson tree[id*2]
#define rson tree[id*2+1]
const int mod=1e9+;
const int maxn=;
const int C=~(<<),M=(<<)-;
int rnd(int last,int &a,int &b)
{
a=(+(last>>))*(a&M)+(a>>);
b=(+(last>>))*(b&M)+(b>>);
return (C&((a<<)+b))%;
}
int n,m,A,B,a[maxn],b[maxn],c[maxn],xs[maxn];
bool cmp(const int& x,const int& y){ return b[x]<b[y]; }
int data[maxn<<],*p;
int* New(int len){ p+=len; return p-len; } //静态的申请空间
void init() //离散化
{
for(int i=;i<n;i++) xs[i]=b[i];
sort(xs,xs+n);
for(int i=;i<n;i++) b[i]=upper_bound(xs,xs+n,b[i])-xs-;
for(int i=;i<n;i++) a[i]=upper_bound(xs,xs+n,a[i])-xs-;
p=data; //在最开始的位置
}
struct Tree
{
int le,ri,d,sum;
int *lid,*rid;
void Set(int v){ d=v; sum=v+; }
int GoLe(int v){ return v==-?v:lid[v]; } //左边有多少<=v的
int GoRi(int v){ return v==-?v:rid[v]; } //右边有多少<=v的
}tree[*maxn];
void pushup(int id){ e.sum=lson.sum+rson.sum; } //取和
void pushdown(int id)
{
if(e.d!=-) //延迟更新
{
lson.Set(e.GoLe(e.d));
rson.Set(e.GoRi(e.d));
e.d=-;
}
}
void Build_tree(int id,int le,int ri)
{
e.le=le,e.ri=ri,e.d=-;
e.lid=New(ri-le+); //左右都要申请空间
e.rid=New(ri-le+);
if(le==ri){ e.sum=(a[le]>=b[le]); return; }
int mid=(le+ri)/;
Build_tree(id*,le,mid);
Build_tree(id*+,mid+,ri);
pushup(id);
int ll=mid-le+,rl=ri-mid;
int *vl=b+le,*vr=b+mid+;
int i=,j=,cnt=;
while(i<ll&&j<rl) //归并排序
{
if(vl[i]<vr[j]) //左边小于右边
{
e.lid[cnt]=i;
e.rid[cnt]=j-;
c[cnt++]=vl[i++];
}
else
{
e.lid[cnt]=i-;
e.rid[cnt]=j;
c[cnt++]=vr[j++];
}
}
while(i<ll) //左边没完的
{
e.lid[cnt]=i;
e.rid[cnt]=j-;
c[cnt++]=vl[i++];
}
while(j<rl) //右边没完的
{
e.lid[cnt]=i-;
e.rid[cnt]=j;
c[cnt++]=vr[j++];
}
int k=;
for(int i=le;i<=ri;i++) b[i]=c[k++];
}
void Update(int id,int x,int y,int v) //更新
{
int le=e.le,ri=e.ri;
if(x<=le&&ri<=y){ e.Set(v); return; } //在区间内
pushdown(id);
int mid=(le+ri)/;
if(x<=mid) Update(id*,x,y,e.GoLe(v)); //左边GoLe代表左边有多少<=v的
if(y>mid) Update(id*+,x,y,e.GoRi(v));//右边同理
pushup(id);
}
int Query(int id,int x,int y) //查询
{
int le=e.le,ri=e.ri;
if(x<=le&&ri<=y) return e.sum; //在区间内直接返回值
pushdown(id); //延迟更新
int mid=(le+ri)/;
int ret=;
if(x<=mid) ret+=Query(id*,x,y); //加上左边
if(y>mid) ret+=Query(id*+,x,y); //加上右边
return ret;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d%d",&n,&m,&A,&B);
for(int i=;i<n;i++) scanf("%d",&a[i]); //输入
for(int i=;i<n;i++) scanf("%d",&b[i]);
init();
Build_tree(,,n-); //建树
int last=,ret=;
for(int i=;i<=m;i++)
{
int l=rnd(last,A,B)%n;
int r=rnd(last,A,B)%n;
int x=rnd(last,A,B)+;
if(l>r) swap(l,r);
if((l+r+x)%) //为奇数是插入
{
x=upper_bound(xs,xs+n,x)-xs-;
Update(,l,r,x);
}
else //否则查询
{
last=Query(,l,r);
ret=(ret+(LL)i*last%mod)%mod;
}
}
printf("%d\n",ret);
}
return ;
}