bzoj 3038: 上帝造题的七分钟2 线段树||hdu 4027
3038: 上帝造题的七分钟2
Time Limit: 3 Sec Memory Limit: 128 MBSubmit: 1066 Solved: 476
[Submit][Status][Discuss]
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
HINT
1:对于100%的数据,1<=n<=100000,1<=l<=r<=n,数列中的数大于0,且不超过1e12。
2:数据不保证L<=R 若L>R,请自行交换L,R,谢谢!
Source
题意:区间开根号,优化:最大值小于1不更新;
#include<bits/stdc++.h> using namespace std; #define ll long long #define pi (4*atan(1.0)) const int N=5e5+10,M=4e6+10,inf=1e9+10; ll sum[N<<2]; ll maxx[N<<2]; ll getnum(ll x) { return (ll)sqrt(x); } void pushup(int pos) { sum[pos]=sum[pos<<1]+sum[pos<<1|1]; maxx[pos]=max(maxx[pos<<1],maxx[pos<<1|1]); } void build(int l,int r,int pos) { if(l==r) { scanf("%lld",&sum[pos]); maxx[pos]=sum[pos]; return; } int mid=(l+r)>>1; build(l,mid,pos<<1); build(mid+1,r,pos<<1|1); pushup(pos); } void update(int L,int R,int l,int r,int pos) { if(l==r) { sum[pos]=getnum(sum[pos]); maxx[pos]=sum[pos]; return; } int mid=(l+r)>>1; if(L<=mid&&maxx[pos<<1]>1) update(L,R,l,mid,pos<<1); if(R>mid&&maxx[pos<<1|1]>1) update(L,R,mid+1,r,pos<<1|1); pushup(pos); } ll query(int L,int R,int l,int r,int pos) { if(L<=l&&r<=R) return sum[pos]; int mid=(l+r)>>1; ll ans=0; if(L<=mid) ans+=query(L,R,l,mid,pos<<1); if(R>mid) ans+=query(L,R,mid+1,r,pos<<1|1); return ans; } int main() { int n,m,i; scanf("%d",&n); build(1,n,1); scanf("%d",&m); while(m--) { int k,l,r; scanf("%d%d%d",&k,&l,&r); if(l>r)swap(l,r); if(k==0) update(l,r,1,n,1); else printf("%lld\n",query(l,r,1,n,1)); } return 0; }