题目描述
题解
开方直接算的时候也许会出来鬼畜的东西,反正我不会。。。
区间直接下放到叶子结点。
优化是如果某一结点权值变成了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);
}
}
}
总结
这里的优化以前没有见过,又多了一种思路。