SqrtTree学习笔记

散步的时候yy区间最值的不同分块做法，发现单点修改\(O(\sqrt{n})\)查询\(O(1)\)的做法不是很会？

于是yy了一个奇怪做法，写出来看看。

考虑查询的时候两端的散点可以用前后缀最值查出来，所以只需要考虑中间的块。

中间这些块似乎比较恶心，不知道怎么做。

于是我们把每一个块的最值拎出来作为一个点，再分一次块，就成功地用\(O(1)\)的时间把问题变成了根号级别的子问题。

于是分块套分块套分块套……，似乎很对？

如果左右端点都在同一个块内那么不是很好玩，就对每一个块里面也分块，也是一个根号级别的子问题。

于是查询\(T(n)=T(\sqrt{n})+O(1)=O(\log \log n)\)。

那么修改的时候呢？要更新这一个块里面的分块、更新总体的分块、更新前后缀，好像就是\(T(n)=2T(\sqrt{n})+\sqrt{n}\)，也就是\(O(\sqrt{n})\)的。

复杂度一脸正确？

然后就被大佬摔在脸上：你这个东西跟sqrt-tree一模一样……

不过hz大佬改了一下，发现只需要分两层，在第二层块数是\(O(n^{\frac 1 4})\)的，于是可以直接暴力维护任意两个块中间的最值。于是就真的是\(O(\sqrt{n})-O(1)\)了。这样预处理的时间变成了\(O(n^{\frac 5 4})\)，但有什么关系呢？反正它除了做模板题以外好像还干不了什么

把hz改后的版本的代码写了出来并且封装了。感觉这东西应该也可以支持\(O(\sqrt{n}+\log n)\)做区间赋值的，打几个标记就可以了，不过懒得写了……（这点应该比正常的SqrtTree优？）

代码以GSS3为例，由于上不了SPOJ所以没有测，但是拍上了。

#include<bits/stdc++.h>
clock_t t=clock();
namespace my_std{
    using namespace std;
    #define pii pair<int,int>
    #define fir first
    #define sec second
    #define MP make_pair
    #define rep(i,x,y) for (int i=(x);i<=(y);i++)
    #define drep(i,x,y) for (int i=(x);i>=(y);i--)
    #define go(x) for (int i=head[x];i;i=edge[i].nxt)
    #define templ template<typename T>
    #define sz 233333
    #define S 400
    typedef long long ll;
    typedef double db;
    mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
    templ inline T rnd(T l,T r) {return uniform_int_distribution<T>(l,r)(rng);}
    templ inline bool chkmax(T &x,T y){return x<y?x=y,1:0;}
    templ inline bool chkmin(T &x,T y){return x>y?x=y,1:0;}
    templ inline void read(T& t)
    {
        t=0;char f=0,ch=getchar();double d=0.1;
        while(ch>'9'||ch<'0') f|=(ch=='-'),ch=getchar();
        while(ch<='9'&&ch>='0') t=t*10+ch-48,ch=getchar();
        if(ch=='.'){ch=getchar();while(ch<='9'&&ch>='0') t+=d*(ch^48),d*=0.1,ch=getchar();}
        t=(f?-t:t);
    }
    template<typename T,typename... Args>inline void read(T& t,Args&... args){read(t); read(args...);}
    char __sr[1<<21],__z[20];int __C=-1,__zz=0;
    inline void Ot(){fwrite(__sr,1,__C+1,stdout),__C=-1;}
    inline void print(register int x)
    {
        if(__C>1<<20)Ot();if(x<0)__sr[++__C]='-',x=-x;
        while(__z[++__zz]=x%10+48,x/=10);
        while(__sr[++__C]=__z[__zz],--__zz);__sr[++__C]='\n';
    }
    void file()
    {
        #ifdef NTFOrz
        freopen("a.in","r",stdin);
        #endif
    }
    inline void chktime()
    {
        #ifndef ONLINE_JUDGE
        cout<<(clock()-t)/1000.0<<'\n';
        #endif
    }
    #ifdef mod
    ll ksm(ll x,int y){ll ret=1;for (;y;y>>=1,x=x*x%mod) if (y&1) ret=ret*x%mod;return ret;}
    ll inv(ll x){return ksm(x,mod-2);}
    #else
    ll ksm(ll x,int y){ll ret=1;for (;y;y>>=1,x=x*x) if (y&1) ret=ret*x;return ret;}
    #endif
//  inline ll mul(ll a,ll b){ll d=(ll)(a*(double)b/mod+0.5);ll ret=a*b-d*mod;if (ret<0) ret+=mod;return ret;}
}
using namespace my_std;

template<typename hh>struct MySqrtTree
{
    struct WTF
    {
        int m;
        vector<hh>a;
        int cnt,B;
        hh mx1[25][25];
        hh mx2[25][25][25];
        vector<int>L,R,id;
        vector<hh>pre,suf;
        void init(int type)
        {
            B=sqrt(m),cnt=(m-1)/B+1;
            L.resize(cnt+5),R.resize(cnt+5),id.resize(m+5);
            pre.resize(m+5),suf.resize(m+5);
            rep(i,1,cnt) L[i]=R[i-1]+1,R[i]=min(L[i]+B-1,m);
            rep(i,1,cnt) rep(j,L[i],R[i]) id[j]=i;
            rep(i,1,cnt) pre[L[i]]=a[L[i]],suf[R[i]]=a[R[i]];
            rep(i,1,cnt) rep(j,L[i]+1,R[i]) pre[j]=pre[j-1]+a[j];
            rep(i,1,cnt) drep(j,R[i]-1,L[i]) suf[j]=a[j]+suf[j+1];
            if (type) return;
            rep(i,1,cnt) rep(j,L[i],R[i]) 
            { 
                hh cur=mx2[i][j-L[i]][j-L[i]]=a[j]; 
                rep(k,j+1,R[i]) 
                cur=cur+a[k],mx2[i][j-L[i]][k-L[i]]=cur; 
            }
            rep(j,1,cnt) { hh cur=mx1[j][j]=mx2[j][0][R[j]-L[j]]; rep(k,j+1,cnt) cur=cur+mx2[k][0][R[k]-L[k]],mx1[j][k]=cur; }
        }
        hh query(int type,int l,int r)
        {
            if (type) return suf[l]+pre[r];
            if (id[l]==id[r]) return mx2[id[l]][l-L[id[l]]][r-L[id[l]]];
            if (id[l]+1==id[r]) return suf[l]+pre[r];
            return suf[l]+mx1[id[l]+1][id[r]-1]+pre[r];
        }
        void modify(int type,int p,hh w)
        {
            a[p]=w;
            int i=id[p];
            pre[L[i]]=a[L[i]],suf[R[i]]=a[R[i]];
            rep(j,L[i]+1,R[i]) pre[j]=pre[j-1]+a[j];
            drep(j,R[i]-1,L[i]) suf[j]=a[j]+suf[j+1];
            if (type) return;
            rep(j,L[i],R[i]){ hh cur=mx2[i][j-L[i]][j-L[i]]=a[j]; rep(k,j+1,R[i]) cur=cur+a[k],mx2[i][j-L[i]][k-L[i]]=cur; }
            rep(j,1,cnt) { hh cur=mx1[j][j]=mx2[j][0][R[j]-L[j]]; rep(k,j+1,cnt) cur=cur+mx2[k][0][R[k]-L[k]],mx1[j][k]=cur; }
        }
    }s[S];
    void init(hh *a,int n)
    {
        s[0].a.resize(n+1);
        s[0].m=n;
        rep(i,1,n) s[0].a[i]=a[i];
        s[0].init(1);
        rep(i,1,s[0].cnt) 
        { 
            s[i].a.resize(s[0].R[i]-s[0].L[i]+3); 
            rep(j,s[0].L[i],s[0].R[i]) s[i].a[j-s[0].L[i]+1]=a[j],++s[i].m; 
        }
        rep(i,1,s[0].cnt) s[i].init(0);
        int cnt=s[0].cnt;
        s[cnt+1].a.resize(cnt+1);
        rep(i,1,cnt) s[cnt+1].a[i]=s[0].pre[s[0].R[i]]; 
        s[cnt+1].m=cnt;
        s[cnt+1].init(0);
    }
    hh query(int l,int r)
    {
        int il=s[0].id[l],ir=s[0].id[r];
        if (il==ir-1) return s[0].query(1,l,r);
        if (il==ir) return s[il].query(0,l-s[0].L[il]+1,r-s[0].L[il]+1);
        return s[0].suf[l]+s[s[0].cnt+1].query(0,il+1,ir-1)+s[0].pre[r];
    }
    void modify(int p,hh w)
    {
        s[0].modify(1,p,w);
        int i=s[0].id[p];
        s[i].modify(0,p-s[0].L[i]+1,w);
        s[s[0].cnt+1].modify(0,i,s[0].pre[s[0].R[i]]);
    }
};

int n,m;
int a[sz];
struct hhh
{
    ll lmx,rmx,mx,sum;
    hhh (ll Lmx=0,ll Rmx=0,ll Mx=0,ll Sum=0){lmx=Lmx,rmx=Rmx,mx=Mx,sum=Sum;}
    const hhh operator + (const hhh &x) const
    {
        hhh ret;
        ret.lmx=max(lmx,x.lmx+sum);
        ret.rmx=max(x.rmx,rmx+x.sum);
        ret.mx=max({mx,x.mx,rmx+x.lmx});
        ret.sum=sum+x.sum;
        return ret;
    }
}A[sz];
MySqrtTree<hhh>tr;

int main()
{
    file();
    read(n);
    rep(i,1,n) read(a[i]),A[i]=hhh(a[i],a[i],a[i],a[i]);
    tr.init(A,n);
    read(m);
    int x,y,z;
    while (m--)
    {
        read(z,x,y);
        if (z==0) tr.modify(x,hhh(y,y,y,y));
        else printf("%lld\n",tr.query(x,y).mx);
    }
    return 0;
}

（正经的学习笔记在路上了……）

转载于:https://www.cnblogs.com/p-b-p-b/p/11397441.html