[BZOJ2161]布娃娃（扫描线+线段树）

题目描述

传送门

题目大意：第i个布娃娃有一个耐心值P[i]以及一个魅力值C[i]，并且还有能够忍受的耐心值的上限R[i]以及下限L[i]。当一个布娃娃j满足L[j] ≤ P[i]并且P[i] ≤ R[j]，那么布娃娃j喜欢布娃娃i。求第i个布娃娃想知道喜欢它的布娃娃中，魅力值第i大的布娃娃的魅力值是多少。

题解

将C离散，然后PLR实际上是一个东西可以看作一维
将区间拆成左端点和右端点，将所有的点排序，从左向右扫。建立权值线段树，扫到区间的左端点就在其对应的C上+1，右端点-1，查询点的话就查询当前权值线段树里的第k大值
注意对于某一个坐标的顺序是加、统计答案、减
查询第k大值的时候要注意一下要保证k>0

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
using namespace std;
#define N 100005
#define Mod 19921228
 
int n,add,first,mod,prod,LSH,cnt,ans;
int P[N],C[N],L[N],R[N],lsh[N];
struct data{int x,val,ty;}opt[N*3];
int sum[N*4];
 
void trans(int *a)
{
    a[1]=first%mod;
    for (int i=2;i<=n;++i)
        a[i]=((long long)a[i-1]*prod+(long long)add+(long long)i)%mod;
}
int cmp(data a,data b)
{
    return a.x<b.x||(a.x==b.x&&a.ty>b.ty);
}
void update(int now)
{
    sum[now]=sum[now<<1]+sum[now<<1|1];
}
void change(int now,int l,int r,int x,int v)
{
    int mid=(l+r)>>1;
    if (l==r)
    {
        sum[now]+=v;
        return;
    }
    if (x<=mid) change(now<<1,l,mid,x,v);
    else change(now<<1|1,mid+1,r,x,v);
    update(now);
}
int query(int now,int l,int r,int k)
{
    if (k<=0) return 0;
    int mid=(l+r)>>1;
    if (l==r) return l;
    int t=sum[now<<1];
    if (t>=k) return query(now<<1,l,mid,k);
    else return query(now<<1|1,mid+1,r,k-t);
}
int main()
{
    scanf("%d",&n);
    scanf("%d%d%d%d",&add,&first,&mod,&prod);trans(P);
    scanf("%d%d%d%d",&add,&first,&mod,&prod);trans(C);
    scanf("%d%d%d%d",&add,&first,&mod,&prod);trans(L);
    scanf("%d%d%d%d",&add,&first,&mod,&prod);trans(R);
    for (int i=1;i<=n;++i)
    {
        if (L[i]>R[i]) swap(L[i],R[i]);
        lsh[++LSH]=C[i];
    }
    sort(lsh+1,lsh+LSH+1);LSH=unique(lsh+1,lsh+LSH+1)-lsh-1;
    for (int i=1;i<=n;++i)
    {
        C[i]=lower_bound(lsh+1,lsh+LSH+1,C[i])-lsh;
        opt[++cnt].x=P[i],opt[cnt].val=i;
        opt[++cnt].x=L[i],opt[cnt].val=C[i],opt[cnt].ty=1;
        opt[++cnt].x=R[i],opt[cnt].val=C[i],opt[cnt].ty=-1;
    }
    sort(opt+1,opt+cnt+1,cmp);
    for (int i=1,j;i<=cnt;i=j)
    {
        j=i;
        while (opt[j].x==opt[i].x)
        {
            if (opt[j].ty) change(1,1,LSH,opt[j].val,opt[j].ty);
            else ans=(ans+lsh[query(1,1,LSH,sum[1]-opt[j].val+1)])%Mod;
            ++j;
        }
    }
    printf("%d\n",ans);
}