洛谷P4145 上帝造题的⑦minutes ②

又是线段树。

区间开平方求和，套路题。

如果开到了1就不用再开下去了，否则直接到底。

记得 l > r 时交换 l r

#include <cstdio>
#include <cmath>
#include <algorithm>
typedef long long LL;
const int N = 100010;
struct SegmentTree {
    LL sum[N << 2];
    bool vis[N << 2];

    inline void pushup(int l, int r, int o) {
        if(l == r) {
            if(sum[o] == 1) {
                vis[o] = 1;
            }
            return;
        }
        sum[o] = sum[o << 1] + sum[o << 1 | 1];
        vis[o] = (sum[o] == (r - l + 1));
        return;
    }
    void change(int L, int R, int l ,int r, int o) {
        if(vis[o]) {
            return;
        }
        int mid = (l + r) >> 1;
        if(l == r) {
            sum[o] = (LL)(sqrt((long double)(sum[o])));
            pushup(l, r, o);
            return;
        }
        if(L <= mid) {
            change(L, R, l, mid, o << 1);
        }
        if(mid < R) {
            change(L, R, mid + 1, r, o << 1 | 1);
        }
        pushup(l, r, o);
        return;
    }
    LL ask(int L, int R, int l, int r, int o) {
        if(L <= l && r <= R) {
            return sum[o];
        }
        if(r < L || R < l) {
            return 0;
        }
        int mid = (l + r) >> 1;
        return ask(L, R, l, mid, o << 1) + ask(L, R, mid + 1, r, o << 1 | 1);
    }
    void build(int l, int r, int o) {
        if(l == r) {
            scanf("%lld", &sum[o]);
            pushup(l, r, o);
            return;
        }
        int mid = (l + r) >> 1;
        build(l, mid, o << 1);
        build(mid + 1, r, o << 1 | 1);
        pushup(l, r, o);
        return;
    }
}SgT;

int main() {
    int n;
    scanf("%d", &n);
    SgT.build(1, n, 1);
    int m;
    scanf("%d", &m);
    int f, x, y;
    while(m--) {
        scanf("%d%d%d", &f, &x, &y);
        if(x > y) {
            std::swap(x, y);
        }
        if(f) {
            printf("%lld
",SgT.ask(x, y, 1, n, 1));
        }
        else {
            SgT.change(x, y, 1, n, 1);
        }
    }
    return 0;
}