LibreOJ #6278. 数列分块入门 2

题目链接：

https://loj.ac/problem/6278

题意：

给你一个你$n$个整数的序列，让你进行两种操作

• 将$[l,r]$内的数字加上一个固定值x
• 询问$[l,r]$中，小于$x^2$的数字的个数

分析：

       这里用分块；首先要预处理，将每一块的元素放入另一个容器内进行排序，当想要查询和修改整个的块的时候，我们对这个专门用于查询修改完整块的容器进行操作，因为我们使这个容器内的每一块元素都是有序的，查询和修改时就很方便；然后查询修改对于那些不完整块的元素，就直接在原数组上暴力解决；
       对于修改操作，对于元素组进行暴力修改，并将修改过后的整个的块的元素放入之前说的那个容器，并进行块元素的排序。中间完整的区间则用数组$a[x]$标记，表示$x$位置的块内的数加上的一个值；
       对于查询操作，对于区间两端不完整块对原数组进行暴力统计，对于完整块，对容器进行二分；

代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
#include <vector>
#include <queue>
#include <stack>
#include <cmath>
#include <set>
#include <map>

using namespace std;

#define inf 0x7f7f7f7f
#define maxn 200050
#define N 100100
#define P 2

typedef long long ll;
typedef struct {
    int u, v, next, w;
} Edge;
Edge e[2];
int cnt, head[1];

inline void add(int u, int v, int w) {
    e[cnt].u = u;
    e[cnt].v = v;
    e[cnt].w = w;
    // e[cnt].f=f;
    e[cnt].next = head[u];
    head[u] = cnt++;
    e[cnt].u = v;
    e[cnt].v = u;
    e[cnt].w = w;
    //    e[cnt].f=-f;
    e[cnt].next = head[v];
    head[v] = cnt++;
}

inline void write(int x) {
    if (x < 0)
        putchar('-'), x = -x;
    if (x > 9)
        write(x / 10);
    putchar(x % 10 + '0');
}

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

int a[50050], opt, l, r, c, n, pos[50050], res, mark[50050], b[50050];
vector<int> vec[250];
int main() {
    cin >> n;
    int sz = sqrt(n);
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        pos[i] = (i - 1) / sz;
        vec[pos[i]].push_back(a[i]);
    }
    for (int i = 0; i <= pos[n]; i++) sort(vec[i].begin(), vec[i].end());
    for (int i = 1; i <= n; i++) {
        scanf("%d%d%d%d", &opt, &l, &r, &c);
        int bl = pos[l], br = pos[r];
        if (!opt) {
            for (int j = l; j <= min((bl + 1) * sz, r); j++) a[j] += c;
            for (int j = 0; j < vec[bl].size(); j++) vec[bl][j] = a[bl * sz + j + 1];
            sort(vec[bl].begin(), vec[bl].end());
            if (bl != br) {
                for (int j = br * sz + 1; j <= r; j++) a[j] += c;
                for (int j = 0; j < vec[br].size(); j++) vec[br][j] = a[br * sz + 1 + j];
                sort(vec[br].begin(), vec[br].end());
            }
            for (int j = bl + 1; j < br; j++) mark[j] += c;

        } else {
            res = 0;
            if (bl == br) {
                for (int j = l; j <= r; j++)
                    if (a[j] + mark[pos[j]] < c * c)
                        res++;
            } else {
                for (int j = l; j <= pos[l] * sz + sz; j++)
                    if (a[j] + mark[pos[j]] < c * c)
                        res++;
                for (int j = pos[r] * sz + 1; j <= r; j++)
                    if (a[j] + mark[pos[j]] < c * c)
                        res++;
                for (int j = bl + 1; j < br; j++)
                    res += lower_bound(vec[j].begin(), vec[j].end(), c * c - mark[j]) - vec[j].begin();
            }
            printf("%d\n", res);
        }
    }
    return 0;
}

我们坚持一件事情，并不是因为这样做了会有效果，而是坚信，这样做是对的。
——哈维尔