花神喜欢步行游历各国,顺便虐爆各地竞赛。花神有一条游览路线,它是线型的,也就是说,所有游历国家呈一条线的形状排列,花神对每个国家都有一个喜欢程度(当然花神并不一定喜欢所有国家)。
每一次旅行中,花神会选择一条旅游路线,它在那一串国家中是连续的一段,这次旅行带来的开心值是这些国家的喜欢度的总和,当然花神对这些国家的喜欢程序并不是恒定的,有时会突然对某些国家产生反感,使他对这些国家的喜欢度开根号
(可能是花神虐爆了那些国家的 OI,从而感到乏味)。
现在给出花神每次的旅行路线,以及开心度的变化,请求出花神每次旅行的开心值。
输入格式
第一行是一个整数 N,表示有 N 个国家;
第二行有 N 个空格隔开的整数,表示每个国家的初始喜欢度
第三行是一个整数 M,表示有 M 条信息要处理;
第四行到最后,每行三个整数 x,l,r,当 x=1时询问游历国家 ll到 r的开心值总和。当 x=2时国家 l 到 r 中每个国家的喜欢度 开根号。
输出格式
每次 x=1x=1 时,每行一个整数。表示这次旅行的开心度。
样例
Input Output
4
1 100 5 5
5
1 1 2
2 1 2
1 1 2
2 2 3
1 1 4
101
11
11
思路:因为这个题数值范围不大所以我们可以吧区间修改当成单点修改来做,比较难的就是如何判断当前这个区间是否开根号不会再发生改变,其实我们只要再线段树中加一个最大值变量就可以了,当这个区间的最大值小于等于1的时候就说明这个区间每个数开根号都不会再发生改变了,直接返回就可以。
代码
#include<stdio.h>
#include<math.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+10;
ll s[N];
struct node
{
ll l,r,t,maxx;
}tr[N];
void push_up(node &u,node &l,node &r)
{
u.t=l.t+r.t;
u.maxx=max(l.maxx,r.maxx);
}
void push_up(int u)
{
push_up(tr[u],tr[2*u],tr[2*u+1]);
}
void build(ll u,ll l,ll r)
{
tr[u].l=l;
tr[u].r=r;
if(l==r)
{
tr[u].t=s[l];
tr[u].maxx=s[l];
}
else
{
ll mid=(l+r)/2;
build(2*u,l,mid);
build(2*u+1,mid+1,r);
push_up(u);
}
}
void update(ll u,ll L,ll R)
{
if(tr[u].maxx<=1) return ;//当最大值小于等于1时直接返回
else if(tr[u].l==tr[u].r)//单点修改
{
tr[u].t=sqrt(tr[u].t);
tr[u].maxx=sqrt(tr[u].maxx);
return;
}
else
{
int mid=(tr[u].l+tr[u].r)/2;
if(L<=mid) update(2*u,L,R);
if(R>mid) update(2*u+1,L,R);
push_up(u);
}
}
node query(ll u,ll L,ll R)
{
if(tr[u].l>=L&&tr[u].r<=R) return tr[u];
else
{
int mid=(tr[u].l+tr[u].r)/2;
if(R<=mid) return query(2*u,L,R);
else if(L>mid) return query(2*u+1,L,R);
else
{
node res,l,r;
l=query(2*u,L,R);
r=query(2*u+1,L,R);
push_up(res,l,r);
return res;
}
}
}
int main()
{
ll n,m,i,j;
scanf("%lld",&n);
for(i=1;i<=n;i++)
{
scanf("%d",&s[i]);
}
build(1,1,n);
scanf("%lld",&m);
for(i=0;i<m;i++)
{
ll a,l,r;
scanf("%lld %lld %lld",&a,&l,&r);
if(a==1)
{
node pp=query(1,l,r);
printf("%lld\n",pp.t);
}
else
{
update(1,l,r);
}
}
}