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

(正經的學習筆記在路上了……)


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM