HDU-4027-Can you answer these queries?

链接：https://vjudge.net/problem/HDU-4027#author=0

题意：

给n个数，两种操作，一种对将区间内l数全部变成原来的平方根，向下取整。

一种求区间的和。

思路：

线段数，区间更新，当区间内所有数都为1时不向下更新，即segment[root] == (r - l + 1)时。

代码：

#include <iostream>
#include <memory.h>
#include <vector>
#include <map>
#include <algorithm>
#include <cstdio>
#include <math.h>

using namespace std;

typedef long long LL;

const int MAXN = 1e5 + 10;
LL segment[MAXN * 4];
LL a[MAXN];
int n, m;
int op, l, r;

void Init()
{
    memset(segment, 0, sizeof(segment));
    memset(a, 0, sizeof(a));
}

void Build_Tree(int root, int l, int r)
{
    if (l == r)
    {
        segment[root] = a[l];
        return;
    }
    int mid = (l + r) / 2;
    Build_Tree(root << 1, l, mid);
    Build_Tree(root << 1 | 1, mid + 1, r);
    segment[root] = segment[root << 1] + segment[root << 1 | 1];
}

void Update_Tree(int root, int l, int r, int ql, int qr)
{
    if (ql > r || l > qr)
        return;
    if (segment[root] == (r - l + 1))
        return;
    if (l == r)
    {
        segment[root] = sqrt(segment[root]);
        return;
    }
    int mid = (l + r) / 2;
    Update_Tree(root << 1, l, mid, ql, qr);
    Update_Tree(root << 1 | 1, mid + 1, r, ql, qr);
    segment[root] = segment[root << 1] + segment[root << 1 | 1];
}

LL Query(int root, int l, int r, int ql, int qr)
{
    if (ql > r || l > qr)
        return 0;
    if (ql <= l && r <= qr)
        return segment[root];
    int mid = (l + r) / 2;
    LL res = 0;
    res += Query(root << 1, l, mid, ql, qr);
    res += Query(root << 1 | 1, mid + 1, r, ql, qr);
    return res;
}

int main()
{
    int cnt = 0;
    while (~scanf("%d", &n))
    {
        printf("Case #%d:\n", ++cnt);
        for (int i = 1; i <= n; i++)
            scanf("%lld", &a[i]);
        scanf("%d", &m);
        Build_Tree(1, 1, n);
        while (m--)
        {
            scanf("%d%d%d", &op, &l, &r);
            if (l > r)
                swap(l, r);
            if (op == 0)
                Update_Tree(1, 1, n, l, r);
            else
                printf("%lld\n", Query(1, 1, n, l, r));
        }
        printf("\n");
    }

    return 0;
}