可持久化treap

平衡樹是一個有用的結構,但是常用的splay等由於均攤複雜度和旋轉操作,無法方便的可持久化。

這時候跳出一個treap,他可以可持久化!

treap是一種笛卡爾樹,有兩個值:位置與權。這樣構建出來的treap滿足兩個性質:1、左兒子位置小於當前節點小於右兒子,2、兒子的權值小於父親。容易猜想,若權值隨機,則樹高是O(logn)。

使用類似二叉堆的旋轉操作可以很方便的維護treap,但是還有一種更好的方法。

fhqtreap:平衡樹中有兩個操作非常關鍵:merge和split。其他大多數操作可以通過這兩種操作實現。

(merge表示把兩顆treap左右相接成一個新的treap,split表示把treap以某個元素爲界分成兩段)

不妨先考慮merge(a,b)。若val[a]>val[b],就遞歸合併rs[a]和b,否則合併a和ls[b]。

沒了。。


如果需要可持久化,對每個操作建立新的節點,沒了。。


#include<bits/stdc++.h>
#define N 40005
#define M 4000005
#define ll long long
using namespace std;
int R,C,n,x,y,Rt;
int tg[M],ls[M],rs[M],size[M],val[M],cnt;ll sum[M],Ans;
vector<int>p[N];
int nd(int k=0)
{
	++cnt;
	tg[cnt]=tg[k];
	ls[cnt]=ls[k];
	rs[cnt]=rs[k];
	size[cnt]=size[k];
	val[cnt]=val[k];
	sum[cnt]=sum[k];
	return cnt;
}
void mdy(int k,int x)
{
	tg[k]+=x;
	val[k]+=x;
	sum[k]+=(ll)x*size[k]*(size[k]+1)/2;
}
void dn(int k)
{
	if (ls[k])mdy(ls[k],tg[k]);
	if (rs[k])mdy(rs[k],tg[k]);
	tg[k]=0;
}
void up(int k)
{
	size[k]=size[ls[k]]+size[rs[k]]+1;
	sum[k]=sum[ls[k]]+sum[rs[k]]+(ll)val[k]*size[k]*(size[k]+1)/2
	-(ll)val[k]*size[ls[k]]*(size[ls[k]]+1)/2
	-(ll)val[k]*size[rs[k]]*(size[rs[k]]+1)/2;
}
void build(int &k,int l,int r)
{
	if (l>r) {k=0;return;}
	k=nd();size[k]=r-l+1;
	int mid=l+r>>1;
	build(ls[k],l,mid-1);
	build(rs[k],mid+1,r);
}
void merge(int &k,int a,int b)
{
	if (!a){k=b;return;}
	if (!b){k=a;return;}
	dn(a);dn(b);
	if (val[a]<val[b])
	{
		int t=nd(a);
		merge(rs[t],rs[a],b);
		k=t;
	}
	else
	{
		int t=nd(b);
		merge(ls[t],a,ls[b]);
		k=t;
	}
	up(k);
}
void split(int k,int &a,int &b,int x)
{
	if (!k) {a=0;b=0;return;}
	dn(k);
	/*
	if (size[ls[k]]+1==x)
	{
		a=nd(k);rs[a]=0;up(a);
		b=rs[k];
		return;
	}
	if (size[ls[k]]==x)
	{
		b=nd(k);ls[b]=0;up(b);
		a=ls[k];
		return;
	}*/
	if (size[ls[k]]+1<=x)
	{
		a=nd(k);
		split(rs[k],rs[a],b,x-size[ls[k]]-1);
		up(a);
	}
	else
	{
		b=nd(k);
		split(ls[k],a,ls[b],x);
		up(b);
	}
}
void out(int x) 
{
	if (!x) return;
	dn(x);
	out(ls[x]);
	printf("%d %d %d %d %d   --  %d\n",x,size[ls[x]]+1,ls[x],rs[x],sum[x],val[x]);
	out(rs[x]);
}
int main()
{
	scanf("%d%d%d",&R,&C,&n);
	for (int i=1;i<=n;i++)
	{
		scanf("%d%d",&x,&y);
		/*x=rand()%R+1;
		y=rand()%C+1;*/
		p[x].push_back(y);
	}
	build(Rt,1,C);
	for (int i=1;i<=R;i++)
	{
		mdy(Rt,1);
		for (int j=0;j<p[i].size();j++)
		{
			int a,b,c;
			split(Rt,a,c,p[i][j]);
			split(a,a,b,p[i][j]-1);
			b=nd();size[b]=1;
			merge(Rt,a,b);
			merge(Rt,Rt,c);
		}
		Ans+=sum[Rt];
	}
//	cout<<cnt<<endl;
	printf("%lld\n",(ll)(R*(R+1)/2)*(C*(C+1)/2)-Ans);
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章