[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 19921228

int n,add,first,mod,prod,LSH,cnt,ans;
int P
,C
,L
,R
,lsh
;
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);
}