BZOJ_2161_布娃娃_权值线段树

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的余数是多少。

Input

输入的第一行有一个整数n,代表布娃娃的个数。
输入的第二行有16个用空格隔开的整数
分别代表Padd,Pfirst,Pmod,Pprod,Cadd,Cfirst,Cmod,Cprod,Ladd,Lfirst,Lmod,Lprod,Radd,Rfirst,Rmod,Rprod。
16个参数均为1到100,000,000中的整数。

Output

输出一个整数,代表所有布娃娃谜团答案的和除以19921228的余数。

Sample Input

3
2 3 4 3 1 4 5 2 3 6 9 1 1 2 3 4

Sample Output

4

先离线下来,然后用类似树状数组二维数点的方式把每个布娃娃拆成两个,一个在L处加,一个在R+1处减。
放在一起排序,然后查询时权值线段树上二分即可。
 
代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 200050
#define maxn 100000000
int t[N*35],ls[N*35],rs[N*35],cnt,n,ans,tot;
struct A {
    int l,r,p,c;
}a[N];
struct P {
    int p,flg,id;
    P() {}
    P(int p_,int f_,int id_) :
        p(p_),flg(f_),id(id_) {}
}d[N<<1];
bool cmp(const P &x,const P &y) {
    if(x.p==y.p) return x.flg<y.flg;
    return x.p<y.p;
}
void update(int l,int r,int x,int v,int &p) {
    if(!p) p=++tot;
    t[p]+=v; 
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(x<=mid) update(l,mid,x,v,ls[p]);
    else update(mid+1,r,x,v,rs[p]);
}
int qk(int l,int r,int k,int p) {
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(k<=t[ls[p]]) return qk(l,mid,k,ls[p]);
    else return qk(mid+1,r,k-t[ls[p]],rs[p]);
}
int main() {
    int pad,pfi,pmod,prd,cad,cfi,cmod,crd,lad,lfi,lmod,lrd,rad,rfi,rmod,rrd;
    register int i;
    scanf("%d%d%d%d%d%d%d%d%d%d%d%d%d%d%d%d%d",&n,&pad,&pfi,&pmod,&prd,&cad,&cfi,&cmod,&crd,&lad,&lfi,&lmod,&lrd,&rad,&rfi,&rmod,&rrd);
    a[1].p=pfi%pmod;
    a[1].c=cfi%cmod;
    a[1].l=lfi%lmod;
    a[1].r=rfi%rmod;
    // if(a[1].l>a[1].r) swap(a[1].l,a[1].r);
    // d[++cnt]=P(a[1].l,1,1);
    // d[++cnt]=P(a[1].r+1,2,1);
    // d[++cnt]=P(a[1].p,3,1);
    // printf("i=%d, l=%d r=%d p=%d c=%d\n",1,a[1].l,a[1].r,a[1].p,a[1].c);
    for(i=2;i<=n;i++) {
        a[i].p=(1ll*a[i-1].p*prd+pad+i)%pmod;
        a[i].c=(1ll*a[i-1].c*crd+cad+i)%cmod;
        a[i].l=(1ll*a[i-1].l*lrd+lad+i)%lmod;
        a[i].r=(1ll*a[i-1].r*rrd+rad+i)%rmod;
        // if(a[i].l>a[i].r) swap(a[i].l,a[i].r);
        // printf("i=%d, l=%d r=%d p=%d c=%d\n",i,a[i].l,a[i].r,a[i].p,a[i].c);
        // d[++cnt]=P(a[i].l,1,i);
        // d[++cnt]=P(a[i].r+1,2,i);
        // d[++cnt]=P(a[i].p,3,i);
    }
    for(i=1;i<=n;i++) {
        if(a[i].l>a[i].r) swap(a[i].l,a[i].r);
        d[++cnt]=P(a[i].l,1,i);
        d[++cnt]=P(a[i].r+1,2,i);
        d[++cnt]=P(a[i].p,3,i);
    }
    // puts("-----------------------------------");
    int now=0,root=0;
    sort(d+1,d+cnt+1,cmp);
    for(i=1;i<=cnt;i++) {
        // printf("i=%d, p=%d flg=%d id=%d\n",i,d[i].p,d[i].flg,d[i].id);
        if(d[i].flg==1) {
            now++;
            update(0,maxn,a[d[i].id].c,1,root);
        }else if(d[i].flg==2) {
            now--;
            update(0,maxn,a[d[i].id].c,-1,root);
        }else {
            if(now>=d[i].id) {
                (ans+=qk(0,maxn,now-d[i].id+1,root))%=19921228;
            }
        }
    }
    printf("%d\n",ans);
}

 

posted @ 2018-05-20 09:38  fcwww  阅读(158)  评论(0编辑  收藏  举报