bzoj 3489 A simple rmq problem —— 主席树套线段树

Posted 2021-01-09 zinn

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=3489

题解：http://www.itdaan.com/blog/2017/11/24/9bc46b690756fe252e17fc3ca90aa01.html

在我挣扎一下午时 Narh 早就A了...

于是看看有何不同，发现 add  和 insert 中必须把 ls[x] = ls[y] , rs[x] = rs[y] 写在前面，而不能是修改 rs 则在那里单写一个 ls[x] = ls[y] 什么的，否则过不了样例...

然后 find 和 query 中一定要有一个 if(!x) return 0; ，否则秒 WA ...

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define mid ((l+r)>>1)
using namespace std;
int const xn=1e5+5,xm=xn*20,xy=xm*20;
int n,m,cnt,rt[xn],ls[xm],rs[xm],cnt2,rt2[xm],ls2[xy],rs2[xy],lst[xn],mx[xy];
struct N{int pr,nxt,pos,val;}p[xn];
int rd()
{
    int ret=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=0; ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘)ret=(ret<<3)+(ret<<1)+ch-‘0‘,ch=getchar();
    return f?ret:-ret;
}
bool cmp(N x,N y){return x.pr<y.pr;}
void add(int &x,int y,int l,int r,N t)
{
    x=++cnt2;
    ls2[x]=ls2[y]; rs2[x]=rs2[y];
    mx[x]=max(mx[y],t.val);
    if(l==r)return;
    if(t.pos<=mid)add(ls2[x],ls2[y],l,mid,t);
    else add(rs2[x],rs2[x],mid+1,r,t);
}
void insert(int &x,int y,int l,int r,N t)
{
    x=++cnt;
    ls[x]=ls[y]; rs[x]=rs[y];
    add(rt2[x],rt2[y],0,n+1,t);//!!!  
    if(l==r)return;
    if(t.nxt<=mid)insert(ls[x],ls[y],l,mid,t);
    else insert(rs[x],rs[y],mid+1,r,t);
}
int find(int x,int l,int r,int L,int R)
{
    if(!x)return 0;//!!
    if(l>=L&&r<=R)return mx[x];
    int ret=0;
    if(mid>=L)ret=max(ret,find(ls2[x],l,mid,L,R));
    if(mid<R)ret=max(ret,find(rs2[x],mid+1,r,L,R));
    return ret;
}
int query(int x,int l,int r,int L,int R,int ql,int qr)
{
    if(!x)return 0;//!!
    if(l>=L&&r<=R)return find(rt2[x],0,n+1,ql,qr);
    int ret=0;
    if(mid>=L)ret=max(ret,query(ls[x],l,mid,L,R,ql,qr));
    if(mid<R)ret=max(ret,query(rs[x],mid+1,r,L,R,ql,qr));
    return ret;
}
int main()
{
    n=rd(); m=rd();
    for(int i=1,x;i<=n;i++)
    {
        x=rd();
        p[i].pos=i; p[i].val=x;
        p[i].pr=lst[x]; p[lst[x]].nxt=i;
        lst[x]=i;
    }
    for(int i=1;i<=n;i++)if(!p[i].nxt)p[i].nxt=n+1;
    sort(p+1,p+n+1,cmp);
    
    for(int i=0,t=1;i<=n+1;i++)
    {
        rt[i]=rt[i-1];
        while(p[t].pr==i&&t<=n)insert(rt[i],rt[i],0,n+1,p[t++]);
    }
    for(int i=1,x,y,l,r,lt=0;i<=m;i++)
    {
        x=rd(); y=rd();
        l=min((x+lt)%n+1,(y+lt)%n+1);
        r=max((x+lt)%n+1,(y+lt)%n+1);
        lt=query(rt[l-1],0,n+1,r+1,n+1,l,r);//
        printf("%d
",lt);
    }
    return 0;
}