3038: 上帝造题的七分钟2
Description
XLk觉得《上帝造题的七分钟》不太过瘾,于是有了第二部。
"第一分钟,X说,要有数列,于是便给定了一个正整数数列。
第二分钟,L说,要能修改,于是便有了对一段数中每个数都开平方(下取整)的操作。
第三分钟,k说,要能查询,于是便有了求一段数的和的操作。
第四分钟,彩虹喵说,要是noip难度,于是便有了数据范围。
第五分钟,诗人说,要有韵律,于是便有了时间限制和内存限制。
第六分钟,和雪说,要省点事,于是便有了保证运算过程中及最终结果均不超过64位有符号整数类型的表示范围的限制。
第七分钟,这道题终于造完了,然而,造题的神牛们再也不想写这道题的程序了。"
——《上帝造题的七分钟·第二部》
所以这个神圣的任务就交给你了。
Input
第一行一个整数n,代表数列中数的个数。
第二行n个正整数,表示初始状态下数列中的数。
第三行一个整数m,表示有m次操作。
接下来m行每行三个整数k,l,r,k=0表示给[l,r]中的每个数开平方(下取整),k=1表示询问[l,r]中各个数的和。
Output
对于询问操作,每行输出一个回答。
Sample Input
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 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
Sample Output
19
7
6
7
6
——我是愉快的分隔符——
这道题就是一个加了Lazy Tag的线段树,写的比较烦,但是应该很好想。
代码:
#include<cstdio>
#include<cmath>
#include<iostream>
using namespace std;
const int maxn=100000;
long long num[maxn+5];
long long value[maxn+5];
int next[maxn+5];
int n,m;
inline int lowbit(int x){return x&-x;}
inline void swap(int &x,int &y){x=x+y;y=x-y;x=x-y;}
inline void add(int x,long long val){
for (int i=x;i<=maxn;i+=lowbit(i))
num[i]+=val;
}
inline long long get(int x){
long long ans=0;
for (int i=x;i>0;i-=lowbit(i))
ans+=num[i];
return ans;
}
inline void initNext(){
for (int i=1;i<=n;i++) next[i]=i;
}
inline int getNext(int x){
return next[x]=next[x]==x?x:getNext(next[x]);
}
inline void solve(int l,int r){
if (l>r) swap(l,r);
for (int i=l;i<=r;i++){
if (next[i]!=i){
i=next[i]=getNext(i);
if (i>r) return;
}
if (value[i]==1) continue;
long long temp=value[i];
value[i]=(long long)sqrt(value[i]);
add(i,(long long)value[i]-temp);
if (value[i]==1) next[i]=getNext(i+1);
}
}
int main(){
scanf("%d",&n);
for (int i=1;i<=n;i++){
scanf("%lld",&value[i]);
add(i,value[i]);
}
initNext();
scanf("%d",&m);
int k,l,r;
for (int i=1;i<=m;i++){
int k,l,r;
scanf("%d%d%d",&k,&l,&r);
if (k==0){
solve(l,r);
}else{
if (l>r) swap(l,r);
printf("%lld\n",(long long)get(r)-get(l-1));
}
}
return 0;
}