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
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); }