花神游历各国(洛谷:线段树区间开方)

主要还是注意到两个点,一个是区间最大值为小于等于1的时候再怎么开方都是1所以不用修改,第二个点事数据范围是1e12,开方六次区间最大值就会变为1,当一个区间修改超过六次就返回.

using i64 = long long;
using ll = long long;
constexpr ll inf = 1e18;
struct Info {
    ll sum = 0;
    ll max = -inf;
    void apply(const Info& v) {
        sum = sum + v.sum;
        max = max + v.max;
    }
};
Info operator+(Info a, Info b) {
    Info ans{};
    ans.sum = a.sum + b.sum;
    ans.max = std::max(a.max, b.max);
    return ans;
}

这个题区间开方不好使用懒标记所以只能用普通线段树,每个线段维护三个信息,区间最大值,区间和,区间被修改了几次.

template<class Info>
struct SegmentTree {
    int n;
    std::vector<Info> info;
    SegmentTree() : n(0) {}
    SegmentTree(int n_, Info v_ = Info()) {
        init(n_, v_);
    }
    template<class T>
    SegmentTree(std::vector<T> init_) {
        init(init_);
    }
    void init(int n_, Info v_ = Info()) {
        init(std::vector<Info>(n_, v_));
    }
    template<class T>
    void init(std::vector<T> init_) {
        n = init_.size();
        info.assign(4 << (int)std::log2(n), Info());
        std::function<void(int, int, int)> build = [&](int p, int l, int r) {
            if (r - l == 1) {
                info[p] = init_[l];
                return;
            }
            int m = (l + r) / 2;
            build(2 * p, l, m);
            build(2 * p + 1, m, r);
            pull(p);
        };
        build(1, 0, n);
    }
    void pull(int p) {
        info[p] = info[2 * p] + info[2 * p + 1];
    }
    void modify(int p, int l, int r, int x, const Info& v) {
        if (r - l == 1) {
            info[p] = v;
            return;
        }
        int m = (l + r) / 2;
        if (x < m) {
            modify(2 * p, l, m, x, v);
        }
        else {
            modify(2 * p + 1, m, r, x, v);
        }
        pull(p);
    }
    void modify(int p, const Info& v) {
        modify(1, 0, n, p, v);
    }
    Info rangeQuery(int p, int l, int r, int x, int y) {
        if (l >= y || r <= x) {
            return Info();
        }
        if (l >= x && r <= y) {
            return info[p];
        }
        int m = (l + r) / 2;
        return rangeQuery(2 * p, l, m, x, y) + rangeQuery(2 * p + 1, m, r, x, y);
    }
    Info rangeQuery(int l, int r) {
        return rangeQuery(1, 0, n, l, r);
    }
    
};

jiangly普通线段树的板子是没有区间修改的,我把区间修改单独拿出来讲

    void rangeApply(int p, int l, int r, int x, int y) {
        if (l >= y || r <= x)return;
        //如果一个线段修改超过六次,就返回因为全部都是1了
        //又或者说一个线段的最大值为1
        if (info[p].max <= 1)return;
        if (r - l == 1) {
            info[p].sum = std::sqrt(info[p].sum);
            info[p].max = info[p].sum;
            return;
        }
        int m = (l + r) / 2;
        rangeApply(2 * p, l, m, x, y);
        rangeApply(2 * p + 1, m, r, x, y);
        pull(p);
    }
    void rangeApply(int l, int r) {
        return rangeApply(1, 0, n, l, r);
    }

第一个返回条件,标准的不在修改范围内返回,第二个条件,修改次数大于等于6了,或者区间最大值小于等于1了,返回,第三个条件,当递归到1的时候对该点的值开方,并更新该点的最大值(我就是在这里卡了,忘记更新最大值了),返回,其它线段就往子线段递归,最后pull合并.

int main() {
    int n;
    std::cin >> n;
    std::vector<Info>a(n);
    for (int i = 0; i < n; i++) {
        std::cin >> a[i].sum;
        a[i].max = a[i].sum;
    }
    SegmentTree<Info> tree(a);
    int m;
    std::cin >> m;
    while (m--) {
        ll op, x, y;
        std::cin >> op >> x >> y;
        if (x > y)std::swap(x, y);
        x--;
        if (op == 0) {
            tree.rangeApply(x, y);
        }
        else {
            std::cout<<tree.rangeQuery(x, y).sum<<'\n';
        }
    }
    return 0;
}

 主函数部分的逻辑很简单不多说了.