BZOJ3211[花神游历各国]--线段树

【链接】
bzoj3211

【解题报告】

线段树。

因为 int 范围下一个数最多开5次根号就会变为1。

所以只需要在普通线段树上打一个标记就好了。

#include
#include
#include
#define LL long long
using namespace std;
const int maxn=100005,maxv=400005;
int n,m,a[maxn];
inline char nc()
{
    static char buf[100000],*l,*r;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int Read()
{
    int res=0,f=1; char ch=nc(),cc=ch;
    while (ch<'0'||ch>'9') cc=ch,ch=nc();
    if (cc=='-') f=-1;
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
    return res*f;
}
struct Tree
{
    LL s[maxv];
    bool flg[maxv];
    void Pushup(int d) {s[d]=s[d<<1]+s[(d<<1)+1]; flg[d]=flg[d<<1]&flg[(d<<1)+1];}
    void Build(int d,int L,int R)
    {
        if (L==R) {s[d]=a[L]; flg[d]=0; if (s[d]<2) flg[d]=1; return;}
        int mid=(R-L>>1)+L;
        Build(d<<1,L,mid); Build((d<<1)+1,mid+1,R);
        Pushup(d);
    }
    void Insert(int d,int x,int y,int L,int R)
    {
        if (flg[d]) return;
        if (L==R) {s[d]=sqrt(s[d]); if (s[d]<2) flg[d]=1; return;}
        int mid=(R-L>>1)+L;
        if (y<=mid) Insert(d<<1,x,y,L,mid);
         else if (x>mid) Insert((d<<1)+1,x,y,mid+1,R);
          else Insert(d<<1,x,mid,L,mid),Insert((d<<1)+1,mid+1,y,mid+1,R);
        Pushup(d);
    }
    LL Query(int d,int x,int y,int L,int R)
    {
        if (x==L&&y==R) return s[d];
        int mid=(R-L>>1)+L;
        if (y<=mid) return Query(d<<1,x,y,L,mid);
         else if (x>mid) return Query((d<<1)+1,x,y,mid+1,R);
          else return Query(d<<1,x,mid,L,mid)+Query((d<<1)+1,mid+1,y,mid+1,R);
    }
}tr;
int main()
{
    freopen("3211.in","r",stdin);
    freopen("3211.out","w",stdout);
    n=Read();
    for (int i=1; i<=n; i++) a[i]=Read();
    tr.Build(1,1,n);
    m=Read();
    for (int i=1,z,x,y; i<=m; i++)
    {
        z=Read(); x=Read(); y=Read();
        if (z==1) printf("%lld\n",tr.Query(1,x,y,1,n));
         else tr.Insert(1,x,y,1,n);
    }
    return 0;
}