您的位置:首页 > 理论基础 > 数据结构算法

bzoj3211 花神游历各国 线段树

2017-07-20 19:11 483 查看
题目大意:

维护一个长度为n的序列,支持一下两个操作:

1、区间开根号(向下取整);

2、区间求和。

n<=100000,m<=200000,序列中的数<=10^9且为非负整数。

题目分析:

都这个逼样了还来刷水,我是心大还是脑子进屎呢T_T

区间开根号这个没有什么数据结构维护的了,但是10^9这么大的数开根号5次也就到1了。

所以说线段树维护一下区间和,然后修改就暴力在线段树上改就可以了,对于已经不需要修改的区间打个标记,修改的时候不动就可以了。

时间复杂度O(nlogn),对于所有数修改次数最多不超过5n次,每次修改时间复杂度不超过logn,单次查询logn,所以就是nlogn啦。

没注意非负还T了两发,我真是……

脑子进屎了……

代码如下:

#include <cstdio>
#include <cmath>
#include <algorithm>
#define N 120000
#define ls(c) (c<<1)
#define rs(c) (c<<1|1)
using namespace std;
typedef long long LL;
int n,m;
int a
;
struct segment{
int l,r,mark;
LL sum;
}seg[N<<2];
void build_tree(int c,int l,int r)
{
seg[c].l=l,seg[c].r=r;
if(l==r)
{
seg[c].sum=a[l];
seg[c].mark=a[l]<=1;
return;
}
int mid=l+r>>1;
build_tree(ls(c),l,mid);
build_tree(rs(c),mid+1,r);
seg[c].sum=seg[ls(c)].sum+seg[rs(c)].sum;
seg[c].mark=seg[ls(c)].mark&seg[rs(c)].mark;
}
void update(int c,int l,int r)
{
if(seg[c].mark) return;
if(seg[c].l==seg[c].r)
{
seg[c].sum=int(sqrt(seg[c].sum));
seg[c].mark=seg[c].sum<=1;
return;
}
int mid=seg[c].l+seg[c].r>>1;
if(l<=mid) update(ls(c),l,r);
if(r>mid) update(rs(c),l,r);
seg[c].sum=seg[ls(c)].sum+seg[rs(c)].sum;
seg[c].mark=seg[ls(c)].mark&seg[rs(c)].mark;
}
LL query(int c,int l,int r)
{
if(l<=seg[c].l && r>=seg[c].r) return seg[c].sum;
int mid=seg[c].l+seg[c].r>>1;
if(r<=mid) return query(ls(c),l,r);
if(l>mid) return query(rs(c),l,r);
return query(ls(c),l,r)+query(rs(c),l,r);
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
scanf("%d",&m);
build_tree(1,1,n);
for(int i=1,opt,l,r;i<=m;i++)
{
scanf("%d%d%d",&opt,&l,&r);
if(l>r) swap(l,r);
if(opt==1) printf("%lld\n",query(1,l,r));
else update(1,l,r);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  数据结构