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

Solution

• 直接线段树维护。

• 因为一个数不会被开方几次，我们可以直接暴力修改。

• 如果一个区间中的数都为 0/1 （开方后为自己），就打上一个标记（修改时可直接退出）。

• 时间复杂度 $O(N\ log\ N)$ 。

Code

#include<cstdio>
#include<cctype>
#include<cmath>
using namespace std;
const int N=1e5+5;
int ql,qr;
int a[N];
long long f[N<<2];
bool g[N<<2];
inline int read()
{
    int X=0,w=0; char ch=0;
    while(!isdigit(ch)) w|=ch=='-',ch=getchar();
    while(isdigit(ch)) X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
inline void update(int v)
{
    f[v]=f[v<<1]+f[v<<1|1];
    g[v]=g[v<<1]&&g[v<<1|1];
}
void make(int v,int l,int r)
{
    if(l==r)
    {
        f[v]=a[l];
        if(f[v]<=1) g[v]=true;
        return;
    }
    int mid=(l+r)>>1;
    make(v<<1,l,mid);
    make(v<<1|1,mid+1,r);
    update(v);
}
void change(int v,int l,int r)
{
    if(g[v]) return;
    if(l==r)
    {
        f[v]=sqrt(f[v]);
        if(f[v]<=1) g[v]=true;
        return;
    }
    int mid=(l+r)>>1;
    if(ql<=mid) change(v<<1,l,mid);
    if(qr>mid) change(v<<1|1,mid+1,r);
    update(v);
}
long long find(int v,int l,int r)
{
    if(ql<=l && r<=qr) return f[v];
    int mid=(l+r)>>1;
    long long sum=0;
    if(ql<=mid) sum+=find(v<<1,l,mid);
    if(qr>mid) sum+=find(v<<1|1,mid+1,r);
    return sum;
}
int main()
{
    int n=read();
    for(int i=1;i<=n;i++) a[i]=read();
    make(1,1,n);
    int m=read();
    while(m--)
    {
        int x=read();
        ql=read(),qr=read();
        if(x==2) change(1,1,n); else printf("%lld\n",find(1,1,n));
    }
    return 0;
}