1. 程式人生 > >【分塊】教主的魔法 @洛谷P2801/upcexam3138

【分塊】教主的魔法 @洛谷P2801/upcexam3138

時間限制: 1 Sec 記憶體限制: 128 MB
題目描述
教主最近學會了一種神奇的魔法,能夠使人長高。於是他準備演示給XMYZ資訊組每個英雄看。於是N個英雄們又一次聚集在了一起,這次他們排成了一列,被編號為1、2、……、N。
每個人的身高一開始都是不超過1000的正整數。教主的魔法每次可以把閉區間[L, R](1≤L≤R≤N)內的英雄的身高全部加上一個整數W。(雖然L=R時並不符合區間的書寫規範,但我們可以認為是單獨增加第L(R)個英雄的身高)
CYZ、光哥和ZJQ等人不信教主的邪,於是他們有時候會問WD閉區間 [L, R] 內有多少英雄身高大於等於C,以驗證教主的魔法是否真的有效。
WD巨懶,於是他把這個回答的任務交給了你。
輸入
第1行為兩個整數N、Q。Q為問題數與教主的施法數總和。
第2行有N個正整數,第i個數代表第i個英雄的身高。
第3到第Q+2行每行有一個操作:
(1)若第一個字母為“M”,則緊接著有三個數字L、R、W。表示對閉區間 [L, R] 內所有英雄的身高加上W。
(2)若第一個字母為“A”,則緊接著有三個數字L、R、C。詢問閉區間 [L, R] 內有多少英雄的身高大於等於C。
輸出
對每個“A”詢問輸出一行,僅含一個整數,表示閉區間 [L, R] 內身高大於等於C的英雄數。
樣例輸入
5 3
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4
樣例輸出
2
3
提示
原先5個英雄身高為1、2、3、4、5,此時[1, 5]間有2個英雄的身高大於等於4。教主施法後變為1、2、4、5、6,此時[1, 5]間有3個英雄的身高大於等於4。
對30%的資料,N≤1000,Q≤1000。
對100%的資料,N≤1000000,Q≤3000,1≤W≤1000,1≤C≤1,000,000,000

第一次寫分塊題
將資料分成sqrt(n)塊,對每個塊分別排序。
對於修改操作,區間內完整的塊修改其add陣列的值,不完整的塊暴力修改在區間內的部分,然後對整個塊排序。
對於查詢操作,區間內完整的塊二分查詢C-add[i]在區間內的位置,然後用該位置減去塊的首位置,不完整的塊暴力統計。

#define FILE_PC() freopen("C:\\Users\\hz\\Desktop\\in.txt","r",stdin)
#include <algorithm>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring> using namespace std; const int maxn = 1000005; const int maxdiv = 1005; int a[maxn],b[maxn]; int l[maxdiv],r[maxdiv],add[maxdiv]; int lendiv,n,m; int main() { // FILE_PC(); scanf("%d%d",&n,&m); for(int i=1; i<=n; i++) { scanf("%d",a+i); b[i] = a[i]; } lendiv = (int
)sqrt(n); //分塊長度 int cntdiv = 0,flag = 1; for(int i=1; i<=n; i++) { if(cntdiv==0)l[++cntdiv] = i;//塊的編號從1開始 if(!flag) { l[++cntdiv] = i; flag = 1; } if(i%lendiv==0) { r[cntdiv] = i; flag = 0; } } r[cntdiv] = n; //最後一塊的右區間 for(int i=1; i<=cntdiv; i++) { sort(b+l[i],b+r[i]+1); }//每個塊裡面的元素排序 for(int ca=0; ca<m; ca++) { char s[20]; int ll,rr,cc; scanf("%s%d%d%d",s,&ll,&rr,&cc); if(s[0]=='M') { int ld = (int)(lower_bound(l+1,l+cntdiv+1,ll)-l); if(l[ld]!=ll) { for(int i=ll; i<min(l[ld],rr+1); i++) { a[i]+=cc; } for(int i=l[ld-1]; i<=r[ld-1]; i++) { b[i] = a[i]; } sort(b+l[ld-1],b+r[ld-1]+1);//重新對這個塊排序 } if(l[ld]<rr+1) { int rd = (int)(lower_bound(r+1,r+cntdiv+1,rr)-r); if(rr!=r[rd]) { for(int i=max(ll,r[rd-1]+1); i<=rr; i++) { a[i] += cc; } for(int i=l[rd]; i<=r[rd]; i++) { b[i] = a[i]; } sort(b+l[rd],b+r[rd]+1);//同上 } for(int i=ld; i<=((rr==r[rd])?rd:rd-1); i++) { //完整的塊 add[i]+=cc; } } } else { int ans = 0; int ld = (int)(lower_bound(l+1,l+cntdiv+1,ll)-l);//給定的左區間落在哪一塊 if(l[ld]!=ll) { for(int i=ll; i<min(l[ld],rr+1); i++) { //暴力統計不完整的塊 if(a[i]+add[ld-1]>=cc)ans++; } } if(l[ld]<rr+1) { int rd = (int)(lower_bound(r+1,r+cntdiv+1,rr)-r);//給定的右區間落在哪一塊 if(rr!=r[rd]) { for(int i=max(ll,r[rd-1]+1); i<=rr; i++) { //同上 if(a[i]+add[rd]>=cc)ans++; } } for(int i=ld; i<=((rr==r[rd])?rd:rd-1); i++) { //統計每一個完整的塊 ans += (int)(b+r[i]+1-lower_bound(b+l[i],b+r[i]+1,cc-add[i])); } } printf("%d\n",ans); } } return 0; }