LGP2801 教主的魔法

题目链接 ： P2801 教主的魔法

这是第一次A分块的题

就是模板题了

每个块内排序 每个整块仅需维护整块的修改量

询问操作：

对于边缘块 直接暴力找在[l, r]内 且比给定值大的有几个

对于整块 二分查找不小于 （给定值 - 本块修改量） 的块有多少个

修改操作：

边缘块直接修改

整块在修改量标记上修改

本题细节较多 尤其是修改和询问的范围

【明明是蒟蒻不熟练。。。

附上代码：

 #include <cstdio>
 #include <algorithm>
 #include <cmath>
 using namespace std;
 const int N = 1e6 + ;
 struct Node{
     int bl, id;
     long long x;
 }node[N];
 int n, blo, q, ts;
 long long atag[N];//修改量标记

 bool rule1(Node a, Node b){
     return a.x < b.x;
 }
 //把[a, b]的元素加c
 void add(int a, int b, long long c){
     for(int i = (node[a].bl - ) * blo + ; i <= min(b, min(n, node[a].bl * blo)); i++)
         if(node[i].id >= a)
             node[i].x += c;
     sort(node + (node[a].bl - ) * blo + , node + min(n, node[a].bl * blo) + , rule1);
     if(node[a].bl != node[b].bl)
         for(int i = (node[b].bl - ) * blo + ; i <= min(n, node[b].bl * blo); i++)
             if(node[i].id <= b)
                 node[i].x += c;
     sort(node + (node[b].bl - ) * blo + , node + min(n, node[b].bl * blo) + , rule1);
     for(int i = node[a].bl + ; i <= node[b].bl - ; i++)
         atag[i] += c;
 }
 //二分查找pos块中比a大的元素个数
 int find(int pos, long long a){
     if(a > node[min(n, pos * blo)].x) return ;
     int l = (pos - ) * blo + , r = min(n, pos * blo), mid;
     while(l < r){
         mid = l + ((r - l) >> );
         if(node[mid].x >= a) r = mid;
         else l = mid + ;
     }
     return min(n, pos * blo) - l + ;
 }
 //询问 [a, b]中大于等于c的元素个数
 int query(int a, int b, long long c){
     int ans = ;
     for(int i = (node[a].bl - ) * blo + ; i <= min(b, min(n, node[a].bl * blo)); i++)
         if(node[i].id >= a && node[i].x >= c - atag[node[a].bl])
             ans++; 

     if(node[a].bl != node[b].bl)
         for(int i = (node[b].bl - ) * blo + ; i <= min(n, node[b].bl * blo); i++)
             if(node[i].id <= b && node[i].x >= c - atag[node[b].bl])
                 ans++;

     for(int i = node[a].bl + ; i <= node[b].bl - ; i++)
         ans += find(i, c - atag[i]);
     return ans;
 }

 int main(){
     scanf("%d%d", &n, &q);
     blo = sqrt(n);
     for(int i = ; i <= n; i++) scanf("%d", &node[i].x);
 //输入
     for(int i = ; i <= n; i++){
         node[i].bl = (i - ) / blo + ;
         node[i].id = i;
     }
     for(int i = ; i <= node[n].bl; i++)
         sort(node + (i - ) * blo + , node + min(n, i * blo) + , rule1);
 //块的分配与块内排序
     char op[];
     int x, y, z;
     for(int i = ; i <= q; i++){
         scanf("%s%d%d%d", op, &x, &y, &z);
         if(op[] == 'M') add(x, y, z);
         else printf("%d\n", query(x, y, z));
     }
     return ;
 }