[bzoj3343] 教主的魔法

Posted 2020-07-26 czllgzmzl

　　分块大法。。

 1 #include<cstdio>
 2 #include<iostream>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<cmath>
 6 using namespace std;
 7 const int maxn=1002333,knum=1023;
 8 int sz[knum],l[knum],r[knum],add[knum];
 9 int bel[maxn],mp[maxn];
10 int a[knum][1023];
11 int i,j,k,n,m,kuai;
12  
13 int ra;char rx;
14 inline int read(){
15     rx=getchar(),ra=0;
16     while(rx<\'0\'||rx>\'9\')rx=getchar();
17     while(rx>=\'0\'&&rx<=\'9\')ra*=10,ra+=rx-48,rx=getchar();return ra;
18 }
19 inline int get(int id,int v){
20     if(v<=0)return sz[id];
21     if(v>a[id][sz[id]])return 0;
22     int l=0,r=sz[id]+1,mid;
23     while(l<r)
24         if(a[id][(mid=l+r>>1)]>=v)r=mid;else l=mid+1;
25     return sz[id]-l+1;
26 }
27 inline void insert(int L,int R,int v){
28     register int i;
29     if(bel[L]==bel[R]){
30         int id=bel[L];
31         for(i=L;i<=R;i++)mp[i]+=v;
32         memcpy(a[id]+1,mp+1+l[id],sz[id]<<2);
33         sort(a[id]+1,a[id]+1+sz[id]);
34         return;
35     }
36     int idl=bel[L],idr=bel[R];
37     for(i=L;i<=r[idl];i++)mp[i]+=v;
38     memcpy(a[idl]+1,mp+1+l[idl],sz[idl]<<2),
39     sort(a[idl]+1,a[idl]+1+sz[idl]);
40      
41     for(i=l[idr];i<=R;i++)mp[i]+=v;
42     memcpy(a[idr]+1,mp+1+l[idr],sz[idr]<<2),
43     sort(a[idr]+1,a[idr]+1+sz[idr]);
44      
45     for(i=idl+1;i<idr;i++)add[i]+=v;
46 }
47 inline int query(int L,int R,int v){
48     register int i;int ans=0,idl=bel[L],idr=bel[R];
49     if(idl==idr){
50         for(i=L;i<=R;i++)ans+=mp[i]+add[idl]>=v;
51         return ans;
52     }
53     for(i=L;i<=r[idl];i++)ans+=mp[i]+add[idl]>=v;
54     for(i=l[idr];i<=R;i++)ans+=mp[i]+add[idr]>=v;
55     for(i=idl+1;i<idr;i++)ans+=get(i,v-add[i]);
56     return ans;
57 }
58 int main(){
59     n=read(),m=read(),kuai=min((int)sqrt(n)+5,n);//     kuai=50;
60     for(i=1;i<=n;i++)bel[i]=(i+kuai-1)/kuai;
61     for(i=1;i<=bel[n];i++){
62         l[i]=(i-1)*kuai+1,r[i]=i==bel[n]?n:(l[i]+kuai-1),sz[i]=r[i]-l[i]+1;
63         for(j=l[i];j<=r[i];j++)mp[j]=a[i][j-l[i]+1]=read();
64         sort(a[i]+1,a[i]+1+sz[i]);
65 //      printf("kuai:%d\\n",i);
66 //      for(j=l[i];j<=r[i];j++)printf("%d  ",a[i][j-l[i]+1]);puts("");
67 //      for(j=l[i];j<=r[i];j++)printf("%d  ",mp[j]);puts("");
68     }char s[2];int a,b,c;
69     while(m--){
70         scanf("%s",s);a=read(),b=read(),c=read();
71         if(s[0]==\'A\')printf("%d\\n",query(a,b,c));else insert(a,b,c);
72 //      for(i=1;i<=n;i++)printf("  %d",mp[i]+add[bel[i]]);puts("");
73     }
74     return 0;
75 }

View Code