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;
}

(正经的学习笔记在路上了……)

posted @ 2019-08-22 23:17  p_b_p_b  阅读(1004)  评论(3编辑  收藏  举报