【BZOJ 3211&3038】 花神游历各国 & 上帝造题的七分钟2

【题目链接】

【BZOJ 3211】 点击打开链接

【BZOJ 3038】 点击打开链接

【算法】

线段树

开根操作直接开到叶子节点，注意当区间中所有数都是0或1时，不需要开根

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 100010

int i,n,m,opt,l,r;
long long a[MAXN];

struct SegmentTree
{
        struct Node
        {
                int l,r;
                long long sum;
                bool flag;
        } Tree[MAXN<<];
        inline void update(int index)
        {
                Tree[index].sum = Tree[index<<].sum + Tree[index<<|].sum;
                Tree[index].flag = Tree[index<<].flag & Tree[index<<|].flag;
        }
        inline void build(int index,int l,int r)
        {
                int mid;
                Tree[index].l = l; Tree[index].r = r;
                if (l == r)
                {
                        Tree[index].sum = a[l];
                        if (a[l] ==  || a[l] == ) Tree[index].flag = true;
                        else Tree[index].flag = false;
                        return;
                }
                mid = (l + r) >> ;
                build(index<<,l,mid);
                build(index<<|,mid+,r);
                update(index);
        }
        inline void modify(int index,int l,int r)
        {
                int mid;
                if (Tree[index].flag) return;
                if (Tree[index].l == Tree[index].r)
                {
                        Tree[index].sum = (long long)sqrt(Tree[index].sum);
                        if (Tree[index].sum ==  || Tree[index].sum == ) Tree[index].flag = true;
                        return;
                }
                mid = (Tree[index].l + Tree[index].r) >> ;
                if (mid >= r) modify(index<<,l,r);
                else if (mid +  <= l) modify(index<<|,l,r);
                else
                {
                        modify(index<<,l,mid);
                        modify(index<<|,mid+,r);
                }
                update(index);
        }
        inline long long query(int index,int l,int r)
        {
                int mid;
                if (Tree[index].l == l && Tree[index].r == r) return Tree[index].sum;
                mid = (Tree[index].l + Tree[index].r) >> ;
                if (mid >= r) return query(index<<,l,r);
                else if (mid +  <= l) return query(index<<|,l,r);
                else return query(index<<,l,mid) + query(index<<|,mid+,r);
        }
} T;
template <typename T> inline void read(T &x)
{
    int f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) { if (c == '-') f = -f; }
    for (; isdigit(c); c = getchar()) x = (x << ) + (x << ) + c - '';
    x *= f;
}
template <typename T> inline void write(T x)
{
    if (x < )
    {
        putchar('-');
        x = -x;
    }
    if (x > ) write(x/);
    putchar(x%+'');
}
template <typename T> inline void writeln(T x)
{
    write(x);
    puts("");
}

int main() {

        read(n);
        for (i = ; i <= n; i++) read(a[i]);
        T.build(,,n);
        read(m);
        while (m--)
        {
                read(opt);
                if (opt == )
                {
                        read(l); read(r);
                        writeln(T.query(,l,r));
                } else
                {
                        read(l); read(r);
                        T.modify(,l,r);
                }
        }
        return ;

}