bzoj3038上帝造题的七分钟2

比较经典的题目，给你一串数，每次有两个操作，一个修改，把区间内每个数开方（向下取整），另一种求和。

分析：我们直接线段树肯定超时，因为是每个数开平方，这样时间复杂度就会退化到n^2，我们可以通过优化来解决这种尴尬的局面。

一个比较简单的优化，就是当这个点的权值已经为0或1时，明显这个节点的值不会改动，那么给这个节点打上标记，而且这个节点的所有子树也不用遍历。
速度还行，500ms多一点。。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const int N=1e5+5;
int n,m;
struct node
{
    ll sum;
    int l,r;
    bool flag;
}t[N*10];
ll a[N];
inline void build(int x,int l,int r)
{
    t[x].l=l,t[x].r=r;
    if (l==r)
    {
        t[x].sum=a[l];
        if (a[l]==1||a[l]==0)t[x].flag=1;
        return;
    }
    int mid=(l+r)>>1;
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
    t[x].flag=t[x<<1].flag&t[x<<1|1].flag;
}
inline void change(int x,int l1,int r1)
{
    int l=t[x].l,r=t[x].r;
    if (t[x].flag)return;
    if (l==r)
    {
        t[x].sum=(ll)sqrt(t[x].sum);
        if (t[x].sum==1||t[x].sum==0)t[x].flag=1;
        return;
    }
    int mid=(l+r)>>1;
    if (r1<=mid)change(x<<1,l1,r1);
    else if(l1>mid)change(x<<1|1,l1,r1);
    else change(x<<1,l1,mid),change(x<<1|1,mid+1,r1);
    t[x].sum=t[x<<1].sum+t[x<<1|1].sum;
    t[x].flag=t[x<<1].flag&t[x<<1|1].flag;
}
inline ll query(int x,int l1,int r1)
{
    int l=t[x].l,r=t[x].r;
    if (l==l1&&r==r1)
    {
        return t[x].sum;
    }
    int mid=(l+r)>>1;
    //printf("%d %d %d %d %d\n",l,r,l1,r1,mid);
    if (r1<=mid)return query(x<<1,l1,r1);
    else if (l1>mid)return query(x<<1|1,l1,r1);
    else
    {
        return query(x<<1,l1,mid)+query(x<<1|1,mid+1,r1);
    }
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n)scanf("%lld",&a[i]);
    build(1,1,n);
    scanf("%d",&m);
    fo(i,1,m)
    {
        int id,x,y;
        scanf("%d%d%d",&id,&x,&y);
        if (x>y)swap(x,y);
        if (id==0)
            change(1,x,y);
        else 
        {
            ll ans=query(1,x,y);
            printf("%lld\n",ans);
        }
    }   
    return 0;
}