HDU 4027 Can you answer these queries? [线段树]

题意：给你长度为n的数列，m个操作，有两种操作类型：

①区间[L,R]每个数开根号

②询问区间[L,R]的和

题解：由于每个数能开根号的次数很少，当区间和为区间长度的时候，说明该区间全为1了，所以不必要更新下去，所以我们可以减少更新线段树的复杂度。

注意：输入的L不一定小于R。。。这个地方TLE了好多次。

AC代码：

#include<stdio.h>
#include<string.h>
#include<iostream>
#include<math.h>
#include<algorithm>
using namespace std;
typedef long long ll;
#define N 100005
ll n,m;
ll a[N];
ll tree[N*4];
void build(ll L,ll R,ll root)
{
	if(L==R)
	{
		tree[root]=a[L];
		return ;
	}
	ll mid=L+R>>1;
	build(L,mid,root<<1);
	build(mid+1,R,root<<1|1);
	tree[root]=tree[root<<1]+tree[root<<1|1];
}
void update(ll l,ll r,ll L,ll R,ll root)
{
	if(tree[root]==R-L+1)return ;
	if(L==R)
	{
		tree[root]=sqrt(tree[root]);
		return ;
	}
	ll mid=L+R>>1;
	if(r<=mid)update(l,r,L,mid,root<<1);
	else if(l>mid)update(l,r,mid+1,R,root<<1|1);
	else
	{
		update(l,mid,L,mid,root<<1);
		update(mid+1,r,mid+1,R,root<<1|1);
	}
	tree[root]=tree[root<<1]+tree[root<<1|1];
}
ll query(ll l,ll r,ll L,ll R,ll root)
{
	if(l<=L&&R<=r)
		return tree[root];
	ll mid=L+R>>1;
	if(r<=mid)return query(l,r,L,mid,root<<1);
	else if(l>mid)return query(l,r,mid+1,R,root<<1|1);
	else return query(l,mid,L,mid,root<<1)+query(mid+1,r,mid+1,R,root<<1|1);
}
int main()
{
	ll cas=1;
	while(~scanf("%lld",&n))
	{
		memset(tree,0,sizeof(tree));
		for(ll i=1;i<=n;i++)
			scanf("%lld",&a[i]);
		build(1,n,1);
		printf("Case #%lld:\n",cas++);
		scanf("%lld",&m);
		while(m--)
		{
			ll op,l,r;
			scanf("%lld%lld%lld",&op,&l,&r);
			if(l>r)swap(l,r);
			if(!op)update(l,r,1,n,1);
			else printf("%lld\n",query(l,r,1,n,1));
		}
		printf("\n");
	}
	return 0;
}