LibreOJ #6281. 数列分块入门 5

题目链接：

https://loj.ac/problem/6281

题意：

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

• 将$[l,r]$的之间的每个数字开方
• 查询$[l,r]$的之间所有数字的和

分析：

       一眼看过去，开方运算，这该怎么分块写；然后看了大佬的博客。。
       给定的数字最大范围是$int$，计算发现，它最多开方$5$次之后就会变成$0$或$1$，而对$0$与$1$的开方忽略；所以对于一个块的处理，我们只要看这个块里的元素是不是都变成了0或1就可以了，不是的话我们就对那一整块暴力，因为$int$最多开方$5$次就会变成$0$或$1$，这样时间复杂度就得到了保障，对于不完整块依然暴力，然后维护它所在块中元素是否全为$0$或$1$的标记；

代码：

#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 50050
#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 opt, l, r, c, sz, n, pos[maxn];
int mark[maxn], a[maxn], sum[maxn];
inline void update(int l, int r) {
    int pl = pos[l], pr = pos[r];
    for (int i = l; i <= min(r, (pl + 1) * sz); i++) {
        sum[pl] -= a[i];
        a[i] = sqrt(a[i]);
        sum[pl] += a[i];
    }
    if (pl != pr) {
        for (int i = pr * sz + 1; i <= r; i++) {
            sum[pr] -= a[i];
            a[i] = sqrt(a[i]);
            sum[pr] += a[i];
        }
    }
    for (int i = pl + 1; i < pr; i++) {
        if (!mark[i]) {
            mark[i] = 1;
            for (int j = i * sz + 1; j <= (i + 1) * sz; j++) {
                sum[i] -= a[j];
                a[j] = sqrt(a[j]);
                sum[i] += a[j];
                if (a[j] > 1)
                    mark[i] = 0;
            }
        }
    }
}
inline int query(int l, int r) {
    int res = 0, pl = pos[l], pr = pos[r];
    for (int i = l; i <= min(r, (pl + 1) * sz); i++) res += a[i];
    if (pl != pr) {
        for (int i = pr * sz + 1; i <= r; i++) res += a[i];
    }
    for (int i = pl + 1; i < pr; i++) res += sum[i];
    return res;
}
int main() {
    cin >> n;
    sz = sqrt(n);
    for (int i = 1; i <= n; i++) {
        a[i] = read();
        pos[i] = (i - 1) / sz;
        sum[pos[i]] += a[i];
    }
    for (int i = 1; i <= n; i++) {
        opt = read(), l = read(), r = read(), c = read();
        if (opt) {
            printf("%d\n", query(l, r));
        } else {
            update(l, r);
        }
    }
    return 0;
}

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