HDU 5737 Differencia（归并树）

【题目链接】 http://acm.hdu.edu.cn/showproblem.php?pid=5737

 

【题目大意】

　　给出两个序列a和b，要求实现两个操作：

　　1. 将a序列的一个区间中的所有数改成同一个数

　　2. 查询一个区间内a数组中大于相同下标b数组中的数的数。

【题解】

　　考虑到b数组是不变的，可以在归并树上预处理出b数组中每个元素在树的左右儿子中的排名，在归并树建立时，可以求出每个区间ai>bi的个数，在发生区间修改的时候，在根节点的有序的b数组中二分查找修改值的排名，将信息下传就可以统计每个区间ai>bi的数目，由于操作具有区间性质，因此可以打延迟标记。查询时下传标记统计答案即可。

【代码】

#include <cstdio>
#include <algorithm>
using namespace std;
const int N=100010,M=300000,U=2000000,mod=1e9+7;
int T,n,m,ans,a[N],b[N],A,B,C=~(1<<31),L,R,x,Ans;
int st[M],en[M],v[M],tag[M],l[U],r[U],seq[U],cnt;
void addtag(int x,int p){v[x]=p?p-st[x]+1:0;tag[x]=p;}
void pb(int x){
    if(tag[x]<0)return;
    addtag(x<<1,l[tag[x]]);addtag(x<<1|1,r[tag[x]]);
    tag[x]=-1;
}
void build(int x,int a,int b){
    tag[x]=-1;
    if(a==b){st[x]=++cnt;seq[cnt]=::b[a];en[x]=cnt;v[x]=::a[a]>=::b[a];return;}
    int mid=(a+b)>>1;
    build(x<<1,a,mid),build(x<<1|1,mid+1,b);
    v[x]=v[x<<1]+v[x<<1|1];
    int al=st[x<<1],ar=en[x<<1],bl=st[x<<1|1],br=en[x<<1|1];
    st[x]=cnt+1;
    while(al<=ar&&bl<=br)seq[++cnt]=seq[al]<seq[bl]?seq[al++]:seq[bl++];
    while(al<=ar)seq[++cnt]=seq[al++];
    while(bl<=br)seq[++cnt]=seq[bl++];
    en[x]=cnt;
    al=st[x<<1],bl=st[x<<1|1];
    for(int i=st[x];i<=cnt;i++){
        while(al<=ar&&seq[al]<=seq[i])al++;
        while(bl<=br&&seq[bl]<=seq[i])bl++;
        l[i]=al-1;r[i]=bl-1;
        if(l[i]<st[x<<1])l[i]=0;
        if(r[i]<st[x<<1|1])r[i]=0;
    }
}
void change(int x,int a,int b,int p){
    if(L<=a&&b<=R){addtag(x,p);return;}pb(x);
    int mid=(a+b)>>1;
    if(L<=mid)change(x<<1,a,mid,l[p]);
    if(R>mid)change(x<<1|1,mid+1,b,r[p]);
    v[x]=v[x<<1]+v[x<<1|1];
}
void ask(int x,int a,int b){
    if(L<=a&&b<=R){ans+=v[x];return;}pb(x);
    int mid=(a+b)>>1;
    if(L<=mid)ask(x<<1,a,mid);
    if(R>mid)ask(x<<1|1,mid+1,b);
    v[x]=v[x<<1]+v[x<<1|1];
}
int lower(int x){
    int l=st[1],r=en[1],pos=0;
    while(l<=r){
        int mid=(l+r)>>1;
        if(seq[mid]<=x)pos=mid,l=mid+1;
        else r=mid-1;
    }return pos;
}
int rnd(){
    A=(36969+(ans>>3))*(A&65535)+(A>>16);
    B=(18000+(ans>>3))*(B&65535)+(B>>16);
    return(C&((A<<16)+B))%1000000000;
}
int main(){
    scanf("%d",&T);
    while(T--){
        scanf("%d%d%d%d",&n,&m,&A,&B);
        for(int i=1;i<=n;i++)scanf("%d",a+i);
        for(int i=1;i<=n;i++)scanf("%d",b+i);
        cnt=Ans=ans=0; build(1,1,n);
        for(int i=1;i<=m;i++){
            L=rnd()%n+1,R=rnd()%n+1,x=rnd()+1;
            if(L>R)swap(L,R);
            if((L+R+x)&1)change(1,1,n,lower(x));
            else{
                ans=0; ask(1,1,n);
                Ans=(1LL*ans*i+Ans)%mod;
            }
        }printf("%d
",Ans);
    }return 0;
}