codevs 2492 上帝造题的七分钟 2（线段树）

codevs 2492 上帝造题的七分钟 2

题目描述 Description

　　XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。

　　“第一分钟，X说，要有数列，于是便给定了一个正整数数列。
　　第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。
　　第三分钟，k说，要能查询，于是便有了求一段数的和的操作。
　　第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。
　　第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。
　　第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
　　第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。”
　　——《上帝造题的七分钟·第二部》
　　所以这个神圣的任务就交给你了。

输入描述 Input Description
　　第一行一个整数n，代表数列中数的个数。
　　第二行n个正整数，表示初始状态下数列中的数。
　　第三行一个整数m，表示有m次操作。
　　接下来m行每行三个整数k,l,r，k=0表示给[l,r]中的每个数开平方(下取整)，k=1表示询问[l,r]中各个数的和。
　　UPD：注意数据中有可能l>r，所以遇到这种情况请交换l和r。

输出描述 Output Description
　　对于询问操作，每行输出一个回答。

样例输入 Sample Input
10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

样例输出 Sample Output
19
7
6

数据范围及提示 Data Size & Hint
　　对于30%的数据，1<=n,m<=1000，数列中的数不超过32767。
　　对于100%的数据，1<=n,m<=100000，1<=l,r<=n，数列中的数大于0，且不超过1e12。
　　注意l有可能大于r，遇到这种情况请交换l,r。

思路：线段树，记录某段区间被开方后是否全为1。

题解：

#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
const int maxn=300000;
struct cc{
    long long sum,l,r;
    bool is_one;
}tree[maxn*2];
long long num[maxn];
void update(long long p)
{
    tree[p].sum=tree[p*2].sum+tree[p*2+1].sum;
    if(tree[p*2].is_one==1&&tree[p*2+1].is_one==1)
    {
        tree[p].is_one=1;
    }
    return;
}
void build(long long l,long long r,long long p)
{
    tree[p].l=l;
    tree[p].r=r;
    if(l==r)
    {
        tree[p].sum=num[l];
        return;
    }
    long long mid=(tree[p].l+tree[p].r)/2;
    build(l,mid,p*2);
    build(mid+1,r,p*2+1);
    update(p);
    return;
}
void change(long long l,long long r,long long p)
{
        if(tree[p].is_one==1)
        {
            return;
        }
        if(tree[p].l==tree[p].r)
        {
        tree[p].sum=sqrt(tree[p].sum);
        if(tree[p].sum==1)
        {
            tree[p].is_one=1;
        }
        return; 
        }

    long long mid=(tree[p].l+tree[p].r)/2;
    if(l<=mid)
    {
        change(l,r,p*2);
    }
    if(mid+1<=r)
    {
        change(l,r,p*2+1);
    }
    update(p);
    return;
}
long long ask(long long l,long long r,long long p)
{
    long long ans=0;
    if(tree[p].l>=l&&tree[p].r<=r)
    {
        return tree[p].sum;
    }
    long long mid=(tree[p].l+tree[p].r)/2;
    if(l<=mid)
    {
        ans+=ask(l,r,p*2);
    }
    if(mid+1<=r)
    {
        ans+=ask(l,r,p*2+1);
    }
    update(p);
    return ans;
}
int main()
{
    int n,m;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&num[i]);
    }
    build(1,n,1);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        long long k,l,r;
        scanf("%lld%lld%lld",&k,&l,&r);
        if(k==0)
        {
            if(l>r)
            {
            swap(l,r);
            }
            change(l,r,1);
        }
        if(k==1)
        {
            if(l>r)
            {
            swap(l,r);
            }
            printf("%lld\n",ask(l,r,1));
        }
    }
    return 0;
}