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

题目描述

传送门

题目大意：第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 19921228int 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);}