【luogu4145】上帝造题的七分钟2 / 花神游历各国--区间开根-线段树

题目背景

XLk觉得《上帝造题的七分钟》不太过瘾，于是有了第二部。

题目描述

"第一分钟，X说，要有数列，于是便给定了一个正整数数列。

第二分钟，L说，要能修改，于是便有了对一段数中每个数都开平方(下取整)的操作。

第三分钟，k说，要能查询，于是便有了求一段数的和的操作。

第四分钟，彩虹喵说，要是noip难度，于是便有了数据范围。

第五分钟，诗人说，要有韵律，于是便有了时间限制和内存限制。

第六分钟，和雪说，要省点事，于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。

第七分钟，这道题终于造完了，然而，造题的神牛们再也不想写这道题的程序了。"

——《上帝造题的七分钟·第二部》

所以这个神圣的任务就交给你了。

输入格式

第一行一个整数n，代表数列中数的个数。

第二行n个正整数，表示初始状态下数列中的数。

第三行一个整数m，表示有m次操作。

接下来m行每行三个整数k,l,r，

• k=0表示给[l,r]中的每个数开平方(下取整)
• k=1表示询问[l,r]中各个数的和。

数据中有可能l>rl，所以遇到这种情况请交换l和r。

输出格式

对于询问操作，每行输出一个回答。

输入输出样例

输入 #1

10
1 2 3 4 5 6 7 8 9 10
5
0 1 10
1 1 10
1 1 5
0 5 8
1 4 8

输出 #1

19
7
6

说明/提示

对于30%的数据，1≤n,m≤1000，数列中的数不超过32767。

对于100%的数据，1≤n,m≤100000， n1≤l,r≤n，数列中的数大于0，且不超过10^10。

注意l有可能大于r，遇到这种情况请交换l,r。

　　线段树，VIS维护是否区间内的数都为一

代码：

#include<cstdio>
#include<iostream>
#include<cmath>
#define int long long
#define N 100100
using namespace std;
int a[N<<1],tr[N<<2],n,m,vis[N<<2];
void update(int k)
{
	tr[k]=tr[k<<1]+tr[k<<1|1];
	if(vis[k<<1]==1 && vis[k<<1|1]==1)vis[k]=1;
	return;
}
void build(int k,int l,int r)
{
	if(l==r)
	{
		tr[k]=a[l];
		if(tr[k]==1||tr[k]==0)vis[k]=1;
		return;
	}
	int mid=(l+r)>>1;
	build(k<<1,l,mid);
	build(k<<1|1,mid+1,r);
	update(k);
}
void change(int k,int l,int r,int x,int y)
{
	if(vis[k])return;
	if(l>=x&&r<=y&&l==r)
	{
		tr[k]=sqrt(tr[k]);
		if(tr[k]==1||tr[k]==0)vis[k]=1;
		return;
	}
	int mid=(l+r)>>1;
	if(x<=mid)change(k<<1,l,mid,x,y);
	if(y>mid)change(k<<1|1,mid+1,r,x,y);
	update(k);
}
int ask(int k,int l,int r,int x,int y)
{
	if(l>=x&&r<=y)return tr[k];
	int mid=(l+r)>>1,ans=0;
	if(x<=mid)ans+=ask(k<<1,l,mid,x,y);
	if(y>mid)ans+=ask(k<<1|1,mid+1,r,x,y);
	return ans;
}
signed main()
{
	scanf("%lld",&n);
	for(int i=1;i<=n;i++)scanf("%lld",&a[i]);
	build(1,1,n);
	scanf("%lld",&m);
	for(int i=1,opt,l,r;i<=m;i++)
	{
		scanf("%lld%lld%lld",&opt,&l,&r);
		if(l>r)swap(l,r);
		if(opt==0)change(1,1,n,l,r);
		if(opt==1)
		{
			printf("%lld\n",ask(1,1,n,l,r));
		}
	}
	return 0;
}