HDU-4027 線段樹

題目鏈接:http://acm.hdu.edu.cn/showproblem.php?pid=4027

題目大意:

給你一個初始序列a[i],然後有兩個操作，一個是詢問區間的和，一個是將區間裏的每一個數變爲原來的平方根，向下取整。操作數<=100000

思路

給定的數初始最大爲\(2^{63}\),那麼我們最多取6次平方根就可以將其變爲1，變成1之後不管怎麼取都是1，所以考慮使用線段樹維護一個區間最大值，當最大值爲1時這個區間就不用計算了，不爲1時就暴力遞歸更新結點的值。
因爲取平方根的次數很少所以暴力遞歸更新的次數也很少不會超時。
不知道線段樹的可以看看我之前寫的線段樹入門:https://startcraft.cn/post/a7bbcc60.html#more
或者
https://blog.csdn.net/lalala445566/article/details/89440830
該題有一個坑點，數據給的區間可能l>r

AC代碼

#include <iostream>
#include <cstring>
#include <cmath>
#include <cstdio>
using namespace std;
typedef long long ll;
#define wfor(i,j,k) for(i=j;i<k;++i)
#define mfor(i,j,k) for(i=j;i>=k;--i)
// void read(int &x) {
//  char ch = getchar(); x = 0;
//  for (; ch < '0' || ch > '9'; ch = getchar());
//  for (; ch >= '0' && ch <= '9'; ch = getchar()) x = x * 10 + ch - '0';
// }
const int maxn = 1e5 + 5;
ll tree[maxn << 2];//區間和
ll maxnum[maxn << 2];//區間最大值
ll num[maxn];
void push_up(int id)
{
    maxnum[id] = max(maxnum[id << 1], maxnum[id << 1 | 1]);
    tree[id] = tree[id << 1] + tree[id << 1 | 1];
}
void bulid(int l, int r, int id)//建樹
{
    if (l == r)
    {
        tree[id] = num[l];
        maxnum[id] = num[l];
        return ;
    }
    int mid = (l + r) >> 1;
    bulid(l, mid, id << 1);
    bulid(mid + 1, r, id << 1 | 1);
    push_up(id);
}
void updata(int l, int r, int L, int R, int id)//更新
{
    if (maxnum[id] <= 1)//當最大值大於1時就返回不用更新了
    {
        return ;
    }
    if (l == r)//暴力更新結點的值
    {
        tree[id] = sqrt(tree[id]);
        maxnum[id] = tree[id];
        return ;
    }
    int mid = (l + r) >> 1;
    if (mid >= L)
        updata(l, mid, L, R, id << 1);
    push_up(id);
    if (mid < R)
        updata(mid + 1, r, L, R, id << 1 | 1);
    push_up(id);
}
ll query(int l, int r, int L, int R, int id)
{
    if (l >= L && r <= R)
    {
        return tree[id];
    }
    ll ans = 0;//累加答案
    int mid = (l + r) >> 1;
    if (mid >= L)
        ans += query(l, mid, L, R, id << 1);
    if (mid < R)
        ans += query(mid + 1, r, L, R, id << 1 | 1);
    return ans;
}
int main()
{
    std::ios::sync_with_stdio(false);
    int n;
    int casecnt = 0;
    while (cin >> n)
    {
        casecnt++;
        memset(tree, 0, sizeof(tree));
        memset(maxnum, 0, sizeof(maxnum));
        cout << "Case #" << casecnt << ":" << endl;
        int i;
        wfor(i, 1, n + 1)
        {
            cin >> num[i];
        }
        bulid(1, n, 1);
        int m;
        cin >> m;
        wfor(i, 0, m)
        {
            int t, l, r;
            cin >> t >> l >> r;
            if (l > r)
                swap(l, r);
            if (t == 0)
            {
                updata(1, n, l, r, 1);//更新
            } else
            {
                ll ans = query(1, n, l, r, 1);
                cout << ans << endl;
            }
        }
        cout << endl;
    }
    return 0;
}