bzoj 2161: 布娃娃 （扫描线+线段树）

题目描述

传送门

题解

把l[i],r[i]看成是l[i]处加入，r[i]+1处删除。
因为要查询的是第k大的问题，所以我们将c[i]离散化后建立权值线段树即可。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
#define N 400003
#define p 19921228
#define LL long long
using namespace std;
int Padd, Pfirst, Pmod, Pprod, Cadd, Cfirst, Cmod, Cprod, Ladd, Lfirst, Lmod, Lprod, Radd, Rfirst, Rmod, Rprod;
int tr[N*4],sum[N*4],n,a[N],b[N],c[N],P[N],l[N],r[N],cnt;
struct data{
    int id,opt,x,y,t,val;
}q[N];
void init(int a[N],int add,int first,int mod,int prod)
{
    a[1]=first%mod;
    for (int i=2;i<=n;i++)
     a[i]=(LL)((LL)a[i-1]*prod+add+i)%mod;
}
int cmp(data a,data b)
{
    return a.x<b.x||a.x==b.x&&a.opt<b.opt;
}
void update(int now)
{
    tr[now]=tr[now<<1]+tr[now<<1|1];
}
void pointchange(int now,int l,int r,int x,int val)
{
    if (l==r) {
        tr[now]+=val;
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) pointchange(now<<1,l,mid,x,val);
    else pointchange(now<<1|1,mid+1,r,x,val);
    update(now);
}
int find(int now,int l,int r,int k)
{
    if(l==r) return l;
    int mid=(l+r)/2;
    if(tr[now<<1]>=k) return find(now<<1,l,mid,k);
    else return find(now<<1|1,mid+1,r,k-tr[now<<1]);
}
int cmp1(int x,int y){
    return x>y;
}
int main()
{
    freopen("a.in","r",stdin);
    freopen("my.out","w",stdout);
    scanf("%d",&n);
    scanf("%d%d%d%d",&Padd,&Pfirst,&Pmod,&Pprod);
    init(P,Padd,Pfirst,Pmod,Pprod);
    scanf("%d%d%d%d",&Cadd,&Cfirst,&Cmod,&Cprod);
    init(c,Cadd,Cfirst,Cmod,Cprod);
    scanf("%d%d%d%d",&Ladd,&Lfirst,&Lmod,&Lprod);
    init(l,Ladd,Lfirst,Lmod,Lprod);
    scanf("%d%d%d%d",&Radd,&Rfirst,&Rmod,&Rprod);
    init(r,Radd,Rfirst,Rmod,Rprod);
    for (int i=1;i<=n;i++)b[i]=c[i];
    sort(b+1,b+n+1);
    int m=unique(b+1,b+n+1)-b-1;
    for (int i=1;i<=n;i++) {
        if (l[i]>r[i]) swap(l[i],r[i]);
        int t=lower_bound(b+1,b+m+1,c[i])-b; t=m-t+1; 
        ++cnt; q[cnt].opt=1; q[cnt].x=l[i]; q[cnt].y=t; q[cnt].val=1;
        ++cnt; q[cnt].opt=1; q[cnt].x=r[i]+1; q[cnt].y=t; q[cnt].val=-1;
        ++cnt; q[cnt].opt=2; q[cnt].id=i; q[cnt].x=P[i]; 
    }
    reverse(b+1,b+m+1);
    sort(q+1,q+cnt+1,cmp);
    int ans=0;
    for (int i=1;i<=cnt;i++){
        if (q[i].opt==1) pointchange(1,1,m,q[i].y,q[i].val);
        if (q[i].opt==2) 
         if (tr[1]>=q[i].id) ans+=b[find(1,1,m,q[i].id)]%p;
        ans%=p;
    }
    printf("%d\n",ans);
}