怕是道水题。
线段树直接上,注意l可能大于r。
如果区间最大值为1,就不操作了。其余暴力改就好了。
#include<bits/stdc++.h>
#define N 100000
using namespace std;
int n,m,l,r,opt;
long long A[N+1],sum[N*4+1],mx[N*4+1];
int ls[N*4+1],rs[N*4+1];
inline char nc()
{
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline long long read()
{
long long x=0,b=1;
char c=nc();
for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
return x*b;
}
inline void write(long long x)
{
if(x==0)putchar('0');
else
{
char buf[25];
int len=0;
if(x<0)putchar('-'),x=-x;
while(x)buf[++len]=x%10+'0',x/=10;
for(int i=len;i>=1;i--)putchar(buf[i]);
}
putchar('\n');
}
inline void pushup(int rt)
{
sum[rt]=sum[rt*2]+sum[rt*2+1];
mx[rt]=max(mx[rt*2],mx[rt*2+1]);
}
inline void build(int rt,int l,int r)
{
int mid;
ls[rt]=l,rs[rt]=r;
if(l==r)sum[rt]=mx[rt]=A[l];
else mid=(l+r)/2,build(rt*2,l,mid),build(rt*2+1,mid+1,r),pushup(rt);
}
inline void modify(int rt,int l,int r)
{
int tmp;
if(mx[rt]<=1)return;
if(ls[rt]==rs[rt]) tmp=sqrt(sum[rt]),sum[rt]=mx[rt]=tmp;
else
{
int mid=(ls[rt]+rs[rt])/2;
if(r<=mid)modify(rt*2,l,r);
else if(l>mid)modify(rt*2+1,l,r);
else modify(rt*2,l,mid),modify(rt*2+1,mid+1,r);
pushup(rt);
}
}
inline long long qurry(int rt,int l,int r)
{
if(ls[rt]==l&&rs[rt]==r)return sum[rt];
int mid=(ls[rt]+rs[rt])/2;
if(r<=mid)return qurry(rt*2,l,r);
else if(l>mid)return qurry(rt*2+1,l,r);
else return qurry(rt*2,l,mid)+qurry(rt*2+1,mid+1,r);
}
int main()
{
freopen("in.txt","r",stdin);
n=read();for(int i=1;i<=n;i++)A[i]=read();
m=read();build(1,1,n);
for(int i=1;i<=m;i++)
{
opt=read();l=read(),r=read();if(l>r)swap(l,r);
if(opt==0)modify(1,l,r);
if(opt==1)write(qurry(1,l,r));
}
return 0;
}
换了种线段树风格。。。