HDU 4027 线段树 Can you answer these queries?

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

这道题最关键的一个地方就是区间更新!

思路: 一开始看到这题,最先想到的就是单结点更新, 但是,这样做是超时的.其实, 这里有一个地方有点难想到,当某个结点的值为1时,开多少次方都还是1,这样就浪费了很多时间. 所以, 要再加一个标记,如果某个区间里的值全为1, 就不继续往下更新.

以下是AC代码:

#include<iostream>
#include<string>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#include<cctype>
#include<iomanip>

#define MID  (f[rt].l+f[rt].r)>>1

using namespace std;
const int maxn=100000;

struct node{
    __int64 sum;
    int l,r;
    bool k;
}f[maxn<<2];

void PushUp(int rt){
    f[rt].sum=f[rt<<1].sum+f[rt<<1|1].sum;
    f[rt].k = f[rt<<1].k&&f[rt<<1|1].k;///如果两个子结点都不用更新,那自己也不用更新了
}

void build(int L,int R,int rt){
    f[rt].l=L, f[rt].r=R, f[rt].k=false;
    if(L==R){
        scanf("%I64d",&f[rt].sum);
        if(f[rt].sum==1)f[rt].k=true;
        return ;
    }
    int mid=MID;
    build(L,mid,rt<<1);
    build(mid+1,R,rt<<1|1);
    PushUp(rt);
}

void update(int L,int R,int rt){
    if(f[rt].k)return;///不用更新了
    if(f[rt].l==f[rt].r){
        f[rt].sum = sqrt(f[rt].sum*1.0);
        if(f[rt].sum==1) f[rt].k=true;
        return;
    }
    int mid=MID;
    if(L<=mid) update(L,R,rt<<1);
    if(R> mid) update(L,R,rt<<1|1);
    PushUp(rt);
}

__int64 query(int L,int R,int rt){
    if(L<=f[rt].l&&R>=f[rt].r)
        return f[rt].sum;
    __int64 ret=0;
    int mid=MID;
    if(L<=mid) ret+=query(L,R,rt<<1);
    if(R> mid) ret+=query(L,R,rt<<1|1);
    return ret;
}

int main(){
    int n,cas=1;
    while(~scanf("%d",&n)){
        build(1,n,1);
        int m; scanf("%d",&m);
        printf("Case #%d:\n",cas++);
        while(m--){
            int c,x,y; scanf("%d%d%d",&c,&x,&y);
            if(x>y)swap(x,y);///这里一定要注意
            if(c)printf("%I64d\n",query(x,y,1));
            else update(x,y,1);
        }
        puts("");
    }
    return 0;
}