[Luogu] 教主的魔法

https://www.luogu.org/problemnew/show/P2801

分块

对于每一块进行排序存储在另一个数组中

二分查询

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>

using namespace std;

#define gc getchar()

const int N = 1e6 + ;
const int M = 1e3 + ;

int A[N], B[N], Pos[N], Add[M];
int n, Q, cnt, block;

inline int read(){
    int x = , f = ; char c = gc;
    while(c < '' || c > '') {if(c == '-') f = -; c = gc;}
    while(c >= '' && c <= '') x = x *  + c - '', c = gc;
    return x * f;
}

inline void Sort(int x){
    int l = (x - ) * block + , r = min(n, x * block);
    for(int i = l; i <= r; i ++) B[i] = A[i];
    sort(B + l, B + r + );
} 

inline void Sec_G(int x, int y, int z){
    if(Pos[x] == Pos[y]){for(int i = x; i <= y; i ++) A[i] += z; Sort(Pos[x]); return ;}
    else {
        for(int i = x; i <= Pos[x] * block; i ++) A[i] += z;
        for(int i = (Pos[y] - ) * block + ; i <= y; i ++) A[i] += z;
        Sort(Pos[x]); Sort(Pos[y]);
    }
    for(int i = Pos[x] + ; i < Pos[y]; i ++) Add[i] += z;
}

inline int to_calc(int x, int num){
    int l = (x - ) * block, r = min(x * block, n), last = r, ans;
    while(l <= r){
        int mid = (l + r) >> ;
        if(B[mid] < num) ans = mid, l = mid + ;
        else r = mid - ;
    }
    return last - ans + ;
}

inline int Sec_A(int x, int y, int z){
    int answer();
    if(Pos[x] == Pos[y]){for(int i = x; i <= y; i ++) if(A[i] + Add[Pos[x]] >= z) answer ++; return answer;}
    else {
        for(int i = x; i <= Pos[x] * block; i ++) if(A[i] + Add[Pos[x]] >= z) answer ++;
        for(int i = (Pos[y] - ) * block + ; i <= y; i ++) if(A[i] + Add[Pos[y]] >= z) answer ++;
    }
    for(int i = Pos[x] + ; i < Pos[y]; i ++) answer += to_calc(i, z - Add[i]);
    return answer;
} 

int main()
{
    freopen("gg.in", "r", stdin);
    n = read(); Q = read();
    block = sqrt(n);
    for(int i = ; i <= n; i ++) A[i] = read();
    for(int i = ; i <= n; i ++) Pos[i] = (i - ) / block + ;
    for(int i = ; i <= n; i ++) Sort(i);
    if(n % block) cnt = n / block + ;
    else cnt = n / block;
    while(Q --){
        string s; cin >> s;
        int x = read(), y = read(), z = read();
        if(s[] == 'A') Sec_G(x, y, z);
        else cout << Sec_A(x, y, z) << endl;
    }

    return ;
}
/*
5 3
1 2 3 4 5
A 1 5 4
M 3 5 1
A 1 5 4
*/