uoj #228. 基础数据结构练习题

问题：

　　维护一个序列，要求支持区间加法，区间开根，区间求和。

---

　　考虑取根号很快就会取到0，维护区间最大最小值，如果一样就是区间覆盖问题，然后发现TLE了，原因是可能存在一种情况8 9 8 9 8 9。

---

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
#define maxn 100100
#define llg register long long
#define llG register long long
#define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
long long n,m,L,R,Q,ty;

inline int getint()
{
       register  int w=0,q=0; char c=getchar();
       while((c<'0' || c>'9') && c!='-') c=getchar(); if(c=='-') q=1,c=getchar(); 
       while (c>='0' && c<='9') w=w*10+c-'0', c=getchar(); return q ? -w : w;
}

struct node
{
    long long min,max,v,add;
}a[maxn*5];

inline void update(llg o,llg l,llg r)
{
    llg lc=(o<<1),rc=(o<<1)|1;//,mid=(l+r)>>1;
    a[o].min=min(a[lc].min,a[rc].min)+a[o].add;
    a[o].max=max(a[lc].max,a[rc].max)+a[o].add;
    a[o].v=a[lc].v+a[rc].v+a[o].add*(r-l+1);
}

inline void sqr(llg o,llg l,llg r,llg L,llg R,llG add)
{
    llg lc=(o<<1),rc=(o<<1)|1,mid=(l+r)>>1;
    llG VAL=add+a[o].add;
    if (l>=L && r<=R)
    {
        if ((a[o].min==a[o].max) || ((a[o].min+1==a[o].max) && (floor(sqrt(a[o].min+add))+1==floor(sqrt(a[o].max+add)))))
        {
            llG ad=sqrt(a[o].min+add);
            ad-=a[o].min+add;
            a[o].add+=ad;
            a[o].min+=ad; a[o].max+=ad;
            a[o].v+=ad*(r-l+1);
            return ;
        }
        sqr(lc,l,mid,L,R,VAL); sqr(rc,mid+1,r,L,R,VAL);
        update(o,l,r);
        return ;
    }
    if (L<=mid)  sqr(lc,l,mid,L,R,VAL);
    if (R>mid) sqr(rc,mid+1,r,L,R,VAL);
    update(o,l,r);
}

inline long long  sum(llg o,llg l,llg r,llg L,llg R,llG add)
{
    if (L<=l && r<=R) return a[o].v+add*(r-l+1);
    llG ans=0;
    llg lc=(o<<1),rc=(o<<1)|1,mid=(l+r)>>1;
    if (L<=mid) ans+=sum(lc,l,mid,L,R,add+a[o].add);
    if (R>mid) ans+=sum(rc,mid+1,r,L,R,add+a[o].add);
    return ans;
}

inline void build(llg o,llg l,llg r)
{
    if (l==r)
    {
        a[o].min=a[o].max=a[o].v=getint();
        return ;
    }
    build(o<<1,l,(l+r)>>1); build((o<<1)|1,((l+r)>>1)+1,r);
    update(o,l,r);
}

inline void add(llg o,llg l,llg r,llg L,llg R,llG v)
{
    if (L<=l && r<=R)
    {
        a[o].add+=v,a[o].v+=v*(r-l+1);
        a[o].min+=v,a[o].max+=v;
        return ;
    }    
    llg lc=(o<<1),rc=(o<<1)|1,mid=(l+r)>>1;
    if (L<=mid) add(lc,l,mid,L,R,v);
    if (R>mid) add(rc,mid+1,r,L,R,v);
    update(o,l,r);
}

int main()
{
    yyj("a");
    cin>>n>>Q;
    build(1,1,n);
    while (Q--)
    {
        ty=getint(); L=getint(),R=getint();
        if (ty==1) add(1,1,n,L,R,getint());
        else if (ty==2) sqr(1,1,n,L,R,0);
        else printf("%lld
",sum(1,1,n,L,R,0));
    }
    return 0;
}