[BZOJ3038]上帝造题的七分钟2（线段树）

题目描述

传送门

题解

开方直接算的时候也许会出来鬼畜的东西，反正我不会。。。
区间直接下放到叶子结点。
优化是如果某一结点权值变成了0或1，那么它之后就不用修改了；如果某个点的左右儿子都不用修改了，这个点也不用修改了。

代码

#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
using namespace std;
#define LL long long

const int max_n=1e5+5;
const int max_tree=max_n*5;

LL n,m,opt,lrange,rrange,ans;
LL a[max_n];
LL sum[max_tree];
bool flag[max_tree];

inline LL in(){
    LL x=0; char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}

inline void update(int now){
    sum[now]=sum[now<<1]+sum[now<<1|1];
    flag[now]=flag[now<<1]&&flag[now<<1|1];
}

inline void build(LL now,LL l,LL r){
    int mid=(l+r)>>1;
    if (l==r){
        sum[now]=a[l];
        return;
    }
    build(now<<1,l,mid);
    build(now<<1|1,mid+1,r);
    update(now);
}

inline void interval_change(LL now,LL l,LL r,LL lrange,LL rrange){
    LL mid=(l+r)>>1;
    if (l==r){
        sum[now]=(LL)sqrt(sum[now]);
        if (sum[now]==0||sum[now]==1) flag[now]=true;
        return;
    }
    if (lrange<=mid&&!flag[now<<1])
      interval_change(now<<1,l,mid,lrange,rrange);
    if (mid+1<=rrange&&!flag[now<<1|1])
      interval_change(now<<1|1,mid+1,r,lrange,rrange);
    update(now);
}

inline LL query(int now,int l,int r,int lrange,int rrange){
    LL mid=(l+r)>>1,ans=0;
    if (lrange<=l&&r<=rrange) return sum[now];
    if (lrange<=mid)
      ans+=query(now<<1,l,mid,lrange,rrange);
    if (mid+1<=rrange)
      ans+=query(now<<1|1,mid+1,r,lrange,rrange);
    return ans;
}

int main(){
    n=in();
    for (int i=1;i<=n;++i)
      a[i]=in();
    build(1,1,n);
    m=in();
    for (int i=1;i<=m;++i){
        opt=in(); lrange=in(); rrange=in();
        if (lrange>rrange) swap(lrange,rrange);
        if (opt==0) interval_change(1,1,n,lrange,rrange);
        else{
            ans=query(1,1,n,lrange,rrange);
            printf("%lld\n",ans);
        }
    }
}

总结

这里的优化以前没有见过，又多了一种思路。