bzoj3211花神游历各国 线段树

题目链接：戳这里

3211: 花神游历各国

Time Limit: 5 Sec  Memory Limit: 128 MB
Submit: 4119  Solved: 1504
[Submit][Status][Discuss]

Description

Input

Output

每次x=1时，每行一个整数，表示这次旅行的开心度

Sample Input

4

1 100 5 5

5

1 1 2

2 1 2

1 1 2

2 2 3

1 1 4

Sample Output

101

11

11

HINT

对于100%的数据， n ≤ 100000，m≤200000 ,data[i]非负且小于10^9

Source

一道还算简单的线段树。

看到题目中的区间操作和查询很容易让人想到线段树，但区间开方好像不满足区间加和性质？

所以我们必须修改到每个叶节点上，但这么做就会TLE。

观察数据：因为是向下取整，所以经过几次操作后就会有区间变成1，显然全是1的区间我们就没必要继续修改下去了，这样就能大大减少复杂度。

对于线段树的每个节点，维护一个lazy或区间最大值，当lazy大于6或区间最大值<=1的时候就直接返回。

代码：

#include<bits/stdc++.h>#define maxn 100010 using namespace std;typedef long long LL;int read(){char c;int sum=0,f=1;c=getchar();while(c<'0' || c>'9'){if(c=='-')f=-1;c=getchar();}while(c>='0' && c<='9'){sum=sum*10+c-'0';c=getchar();}return sum*f;}int n,ql[maxn];int m;LL sum[maxn*4];int lazy[maxn*4];void pushup(int id){sum[id]=sum[id<<1]+sum[id<<1|1];return;}void build(int id,int l,int r){if(l==r){sum[id]=ql[l];return;}int mid=l+r>>1;build(id<<1,l,mid);build(id<<1|1,mid+1,r);pushup(id);}void update(int id,int ql,int qr,int l,int r){if(l>=ql && r<=qr){lazy[id]++;if(l==r){sum[id]=sqrt(sum[id]);return;}}int mid=(l+r)>>1,lson=(id<<1),rson=(id<<1|1);if(lazy[lson]<6 && ql<=mid) update(lson,ql,qr,l,mid);if(lazy[rson]<6 && qr>mid) update(rson,ql,qr,mid+1,r);pushup(id);return;}LL query(int id,int ql,int qr,int l,int r){if(l>=ql && r<=qr)return sum[id];LL ans=0;    int mid=(l+r)>>1;    if(ql<=mid) ans+=query(id<<1,ql,qr,l,mid);    if(qr>mid) ans+=query(id<<1|1,ql,qr,mid+1,r);    return ans;}int main(){n=read();for(int i=1;i<=n;i++)ql[i]=read();build(1,1,n);m=read();for(int i=1;i<=m;i++){int f=read(),l=read(),r=read();if(f==1) printf("%lld\n",query(1,l,r,1,n));if(f==2) update(1,l,r,1,n);}return 0;}