Can you answer these queries?(线段树)

题目链接:Can you answer these queries?

大致题意

给定一个正整数n, 接下来对于区间[1, n]有如下两种操作:

​ ①对于区间[l, r]里的每一个数, 变为原来的√x
​ ②询问区间[l, r]的和.

解题思路

线段树的区间修改和区间查询.

当我们遇到操作①时, 修改时可以遍历去修改每一个节点. 因为他每次的操作是把当前这个数字开平方. 即使当前数字为2E63, 那么由于每次都向下取整, 也只需要6次就可以将它变为1了. 因此对于修改操作每一个叶子节点也最多会遍历6次. 是不会超时的.

需要注意的点: ①输入的x和y大小不保证x<y ②当某一区间的各个元素都是1后, 不需要再进行修改了.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
const int N = 1E5 + 10;
ll w[N];
struct node {
    int l, r;
    ll val;
}t[N << 2];
void pushdown(node& op) { op.val = sqrt(op.val); }

void pushup(int x) { t[x].val = t[x << 1].val + t[x << 1 | 1].val; }

void build(int l, int r, int x = 1) {
    t[x] = { l, r, w[l] };
    if (l == r) return;
    int mid = l + r >> 1;
    build(l, mid, x << 1), build(mid + 1, r, x << 1 | 1);
    pushup(x);
}

void modify(int l, int r, int x = 1) {
    if (t[x].val == t[x].r - t[x].l + 1) return; //区间元素都是1
    if (t[x].l == t[x].r) { pushdown(t[x]); return; } //叶子节点
    
    int mid = t[x].l + t[x].r >> 1;
    if (l <= mid) modify(l, r, x << 1);
    if (r > mid) modify(l, r, x << 1 | 1);
    pushup(x);
}

ll ask(int l, int r, int x = 1) {
    if (l <= t[x].l && r >= t[x].r) return t[x].val;
    int mid = t[x].l + t[x].r >> 1;
    ll res = 0;
    if (l <= mid) res += ask(l, r, x << 1);
    if (r > mid) res += ask(l, r, x << 1 | 1);
    return res;
}
int main()
{
    int T = 1;
    int n, m;
    while (~scanf("%d", &n)) {
        rep(i, n) scanf("%lld", &w[i]);
        build(1, n);
        
        printf("Case #%d:\n", T++);
        scanf("%d", &m);
        while (m--) {
            int op, l, r; scanf("%d %d %d", &op, &l, &r);
            if (l > r) swap(l, r);
            if (op) printf("%lld\n", ask(l, r));
            else modify(l, r);
        }
        printf("\n");
    }
    return 0;
}

kuangbin线段树专题点这里!!!

END