题意:给定的n个数,我们定义两个操作,分别是区间开根号以及区间求和。共有m次查询,其中n,m≤1e5。
思路:一个1e18以内的数,经过最多6次开平方操作后,会变成1。建立线段树,对于每一个叶子节点我们最多进行6次的更新操作后,每次就不会改变大小了。那么我们可以通过维护区间和,如果区间和等于区间长度则不需要更新(或者区间的最大值和区间和,对于区间最大值是1的区间,我们就可以直接不考虑了)因为对其中所有数做开平方操作,都不会改变区间的数,以及区间和。 对于需要进行开平方的区间,我们一直更新到叶子节点,因为每个叶子节点最多更新6次,所以这个是可以接受的,时间复杂度O(nlogn)。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
const int N = 1e6+5;
ll sum[4*N],a[N];
void build(int l,int r,int rt){
if(l==r){
sum[rt]=a[l];
return;
}
int m = (l+r)>>1;
build(lson);
build(rson);
sum[rt] = sum[rt<<1]+sum[rt<<1|1];
}
void up(int L,int R,int l,int r,int rt){
if(sum[rt]==(r-l+1)) return;
if(l==r){
sum[rt] = (ll)sqrt(sum[rt]);
return;
}
int m = (l+r)>>1;
if(L<=m) up(L,R,lson);
if(R>m) up(L,R,rson);
sum[rt] = sum[rt<<1]+sum[rt<<1|1];
}
ll query(int L,int R,int l,int r,int rt){
if(L<=l&&r<=R) return sum[rt];
int m = (l+r)>>1;
ll ret = 0;
if(L<=m) ret += query(L,R,lson);
if(R>m) ret += query(L,R,rson);
return ret;
}
int main(){
int n,q,cas=1;
while(scanf("%d",&n)!=EOF){
for(int i=1;i<=n;i++) scanf("%lld",&a[i]);
build(1,n,1);
scanf("%d",&q);
printf("Case #%d:\n",cas++);
while(q--){
int c,l,r;
scanf("%d%d%d",&c,&l,&r);
if(l>r) swap(l,r); ///l可能大于r
if(c==0) up(l,r,1,n,1);
else printf("%lld\n",query(l,r,1,n,1));
}
printf("\n");
}
return 0;
}