[BZOJ]2161: 布娃娃 权值线段树

Description

小时候的雨荨非常听话，是父母眼中的好孩子。在学校是老师的左右手，同学的好榜样。后来她成为艾利斯顿第二代考神，这和小时候培养的良好素质是分不开的。雨荨的妈妈也为有这么一个懂事的女儿感到高兴。一次期末考试
，雨荨不知道第多少次，再次考了全年级第一名。雨荨的妈妈看到女儿100分的成绩单时，脸上又泛起了幸福的笑容，作为奖励，她给雨荨买了n个布娃娃。细心的雨荨发现，第i个布娃娃有一个耐心值P[i]以及一个魅力值C[i]，
并且还有能够忍受的耐心值的上限R[i]以及下限L[i]。当一个布娃娃j满足L[j]<=P[i]并且P[i]<=R[j]，那么布娃娃j喜欢布娃娃i。雨荨还发现，一个布娃娃有可能喜欢它自己。每个布娃娃心中都有一个谜团，具体来说就是：第
i个布娃娃想知道喜欢它的布娃娃中，魅力值第i大的布娃娃的魅力值是多少，并且称这个布娃娃的谜团答案为这个魅力值的大小，如果不存在，那么这个布娃娃的谜团答案为0。鉴于雨荨的上司栋栋不让题目的数据过大，下面给
出数据的生成方法：给出16个参数：
Padd, Pfirst, Pmod, Pprod, Cadd, Cfirst, Cmod, Cprod, Ladd, Lfirst, Lmod, Lprod, Radd, Rfirst, Rmod, Rprod。
P[1] = Pfirst % Pmod, P[i] = (P[i-1] Pprod + Padd + i) % Pmod (i > 1)。
对于C、L、R数组也有类似的得到方式, %代表取余运算。注意：L和R数组生成完之后，如果某个布娃娃的忍耐度上限小于下限，那么交换它的上限和下限。当然，雨荨也不会让你告诉她每个布娃娃的谜团答案，因为那样会使输出
数据很大。所以雨荨希望你告诉她，所有布娃娃谜团答案的和除以19921228的余数是多少。

题解：

又看错题了。这题我们可以这么做：把一个布娃娃拆成3个操作：1、加入这个布娃娃。2、删除这个布娃娃。3、统计它的答案。然后，给每个操作一个pos，1是L[i]，2是R[i]，3是P[i]，然后我们给这些操作按照pos排序，然后扫一次，这样每次统计i的答案时，就能保证当前的布娃娃的L都<=P[i]，R都>=P[i]，也就是说喜欢它，然后找第k大在离散化后用权值线段树就可以了。时间复杂度O(nlogn)。

代码：

#include<bits/stdc++.h>using namespace std;#define LL long longconst int Maxn=100010;const LL mod=19921228;int n;LL Padd, Pfirst, Pmod, Pprod, Cadd, Cfirst, Cmod, Cprod, Ladd, Lfirst, Lmod, Lprod, Radd, Rfirst, Rmod, Rprod;LL P[Maxn],C[Maxn],L[Maxn],R[Maxn],v[Maxn];struct Opt{int pos,op,id;}q[Maxn*3];bool cmp(Opt a,Opt b){return(a.pos==b.pos)?a.op>b.op:a.pos<b.pos;}struct Tree{int l,r,lc,rc,c;}tr[Maxn*2];int trlen=0;void build(int l,int r){    int t=++trlen;    tr[t].l=l;tr[t].r=r;tr[t].c=0;    if(l<r)    {        int mid=l+r>>1;        tr[t].lc=trlen+1;build(l,mid);        tr[t].rc=trlen+1;build(mid+1,r);    }}void change(int now,int p,int x){    if(tr[now].l==tr[now].r){tr[now].c+=x;return;}    int lc=tr[now].lc,rc=tr[now].rc,mid=tr[now].l+tr[now].r>>1;    if(p<=mid)change(lc,p,x);    else change(rc,p,x);    tr[now].c=tr[lc].c+tr[rc].c;}int query(int now,int k){    if(tr[now].l==tr[now].r)return tr[now].l;    int lc=tr[now].lc,rc=tr[now].rc,mid=tr[now].l;    if(tr[rc].c>=k)return query(rc,k);    else return query(lc,k-tr[rc].c);}int main(){    scanf("%d",&n);    scanf("%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld%lld",&Padd,&Pfirst,&Pmod,&Pprod,&Cadd,&Cfirst,&Cmod,&Cprod,&Ladd,&Lfirst,&Lmod,&Lprod,&Radd,&Rfirst,&Rmod,&Rprod);    P[1]=Pfirst%Pmod;for(int i=2;i<=n;i++)P[i] = (P[i-1]*Pprod + Padd + (LL)(i)) % Pmod;    C[1]=Cfirst%Cmod;for(int i=2;i<=n;i++)C[i] = (C[i-1]*Cprod + Cadd + (LL)(i)) % Cmod;    L[1]=Lfirst%Lmod;for(int i=2;i<=n;i++)L[i] = (L[i-1]*Lprod + Ladd + (LL)(i)) % Lmod;    R[1]=Rfirst%Rmod;for(int i=2;i<=n;i++)R[i] = (R[i-1]*Rprod + Radd + (LL)(i)) % Rmod;    for(int i=1;i<=n;i++)if(L[i]>R[i])swap(L[i],R[i]);    for(int i=1;i<=n;i++)    {        q[i*3]={P[i],0,i};        q[i*3-1]={L[i],1,i};        q[i*3-2]={R[i],-1,i};        v[i]=C[i];    }    sort(v+1,v+1+n);    for(int i=1;i<=n;i++)C[i]=lower_bound(v+1,v+n+1,C[i])-v;//离散化C    sort(q+1,q+1+3*n,cmp);    LL ans=0;    build(1,n);    for(int i=1;i<=3*n;i++)    {        if(q[i].op)change(1,C[q[i].id],q[i].op);        else if(tr[1].c>=q[i].id)ans=(ans+v[query(1,q[i].id)])%mod;    }    printf("%lld",ans);}