2023集 数据结构4 P4145 上帝造题的七分钟 2 / 花神游历各国

上帝造题的七分钟 2 / 花神游历各国

题目背景

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

题目描述

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

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

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

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

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

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

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

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

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

输入格式

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

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

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

接下来 $m$ 行每行三个整数 k l r。

• $k=0$ 表示给 $[l,r]$ 中的每个数开平方（下取整）。

• $k=1$ 表示询问 $[l,r]$ 中各个数的和。

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

输出格式

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

样例 #1

样例输入 #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\le n,m\le 10^3$，数列中的数不超过 $32767$。

对于 $100\%$ 的数据，$1\le n,m\le 10^5$，$1\le l,r\le n$，数列中的数大于 $0$，且不超过 $10^{12}$。

思路

开方运算不需要tag来传递，因为开方不是通过简单的相加相减所得到的，所以采取ax数组，用于储存最大的节点，如果此时这个节点小于等于1，那么就没有进行开方的必要了，可以直接跳过运算

完整代码

#include<bits/stdc++.h>
using namespace std;
#define int long long int
#define lc(x) (x<<1)
#define rc(x) (x<<1|1)
const int N=1e6+234;
int sm[N<<2],a[N],ax[N];
int n,m;
void pushup(int x)
{
	sm[x]=sm[lc(x)]+sm[rc(x)];
	ax[x]=max(ax[lc(x)],ax[rc(x)]);//寻找更大的用于剪枝
}//更新线段树
void build(int x,int l,int r){
	if(l==r){
		sm[x]=ax[x]=a[l];//记得初始化ax的值
		return;
	}
	int mid=(l+r)>>1;
	build(lc(x),l,mid);
	build(rc(x),mid+1,r);
	pushup(x);
}//建树
void update(int x,int l,int r,int L,int R)
{
	if(l==r)
	{
		sm[x]=sqrt(sm[x]);
		ax[x]=sqrt(ax[x]);
		return;
	}
	if(ax[x]<=1)return;//剪枝
	int mid=(l+r)>>1;
	if(L<=mid) update(lc(x),l,mid,L,R);
	if(R>mid) update(rc(x),mid+1,r,L,R);
	pushup(x);
}//区间修改
int query(int x,int l,int r,int L,int R)
{
	if(L<=l&&r<=R)return sm[x];
	int mid=(l+r)>>1;
	int res=0;
	if(L<=mid)res+=query(lc(x),l,mid,L,R);
	if(R>mid)res+=query(rc(x),mid+1,r,L,R);
	return res;
}//区间查询
signed main(){
	scanf("%lld",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&a[i]);
	}
	build(1,1,n);
	scanf("%lld",&m);//输入
	while(m--){
		int kk,ll,rr;
		scanf("%lld %lld %lld",&kk,&ll,&rr);
		if(ll>rr)swap(ll,rr);
		if(kk==1)
			cout<<query(1,1,n,ll,rr)<<endl;
		if(kk==0)
			update(1,1,n,ll,rr);
	}
	return 0;
}