【BZOJ 3211】花神游历各国 &【 线段树区间开根】

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

题解

对线段树区间开根，听说很简单啊。
其实对一个10^9的数开根，最多开5次，所以可以暴力，并记录这个区间需不需要开根（对于数<= 1就无需开根）

然后就套路一发

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

#define lson o << 1
#define rson o << 1 | 1
#define N 100010
typedef long long LL;

int c[N << 2];
LL sum[N << 2];

void pushup(int o)
{
    sum[o] = sum[lson] + sum[rson];
    c[o] = c[lson] || c[rson] ? 1 : 0;
}

void build(int o,int l,int r)
{
    if(l == r)
    {
        scanf("%lld",&sum[o]);
        c[o] = sum[o] > 1 ? 1 : 0;
        return;
    }
    int mid = (l+r) >> 1;
    build(lson,l,mid);
    build(rson,mid+1,r);
    pushup(o);
}

int ll,rr;
LL query(int o,int l,int r)
{
    if(ll <= l && rr >= r) return sum[o];
    int mid = (l+r) >> 1;
    LL ans = 0;
    if(ll <= mid) ans += query(lson,l,mid);
    if(rr > mid) ans += query(rson,mid+1,r);
    return ans;
}

void update(int o,int l,int r)
{
    if(!c[o]) return;
    if(l == r)
    {
        sum[o] = (int)sqrt(sum[o]);
        if(sum[o] <= 1) c[o] = 0;
        return;
    }
    int mid = (l+r) >> 1;
    if(ll <= mid && c[lson]) update(lson,l,mid);
    if(rr > mid && c[rson])update(rson,mid+1,r);
    pushup(o);
    return;
}

int main()
{
    int n,m,opt;
    scanf("%d",&n);
    build(1,1,n);   
    scanf("%d",&m);
    while(m--)
    {
        scanf("%d%d%d",&opt,&ll,&rr);
        if(opt == 1) printf("%lld\n",query(1,1,n));
        else
            update(1,1,n);
    }
    return 0;
}