大二上的时候,写过一个AVL的操作演示,今天一看Splay,发现和AVL其实一样,加上线段树的基础,懒惰标记什么都知道,学起来轻松许多哦
我参考的模板来自这里 http://blog.csdn.net/u013480600/article/list/2
里面有大量的ch[r][0] ch[r][1]等 我建议用宏定义取代,写的时候方括号少打了很多,等做的题多得时候,我再把自己使用的模板发来
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
/*===============================
Splay树模板
1、tot1,tot2都是从1开始
2、
================================*/
#define key_value ch[ch[root][1]][0]
#define ls(r) ch[r][0]
#define rs(r) ch[r][1]
#define ll long long
#define repe(i,s,e) for(int i=s;i<=e;i++)
#define rep(i,s,e) for(int i=s;i<e;i++)
const int MAXN=100000 +10;
int pre[MAXN],ch[MAXN][2],sz[MAXN],root,tot1;
// 父节点、左右孩子(0为左,同时0为左旋)、子树规模、根节点、结点数量
int key[MAXN];//结点的值
int add[MAXN];
ll sum[MAXN];//sum[i]=v 以i为root的树的和,
int s[MAXN],tot2;/??????
//内存池、内存池容量(这题用不到,如果有删除操作,内存不够可以这样
//s中存的是被回收的内存,从后面可以看到,tot2不为0的时候优先使用s[tot2]=x中的
//pre[x],ch[x],size[x]等,
//否则就使用tot1++之后产生的数x的对应位置
int a[MAXN];//初始时的数组,建树时用
int n,q;
void newnode(int &r,int father, int k)//r必须是&
{
if(tot2)r=s[tot2--];//取得时候tot2--,存++tot2
else r=++tot1;
pre[r]=father;
sz[r]=1;
key[r]=k;
add[r]=sum[r]=0;
ch[r][0]=ch[r][1]=0;
}
void updateadd(int r, int av)
{
if(!r)return;//??
add[r]+=av;
key[r]+=av;
sum[r]+=(ll)av*sz[r];
}
//通过孩子结点更新父亲结点
void pushup(int r)
{
sz[r]=sz[ls(r)]+sz[rs(r)]+1;
sum[r]=sum[ls(r)]+sum[rs(r)]+key[r];
}
//将延迟标记更新到孩子结点
void pushdown(int r)
{
if(add[r])
{
updateadd(ls(r),add[r]);
updateadd(rs(r),add[r]);
add[r]=0;
}
}
//建树区间[l,r],先建立中间结点,再建两端,
//注意和线段树的区别
void build(int &x, int l, int r, int father)
{
if(l>r)return;
int mid=(l+r)/2;
newnode(x, father, a[mid]);
build(ch[x][0], l, mid-1, x);
build(ch[x][1], mid+1, r, x);
pushup(x);
}
//初始化,前后各加一个king结点
void Init()
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
root=tot1=tot2=0;
ls(root)=rs(root)=pre[root]=sz[root]=add[root]=sum[root]=key[root]=0;
newnode(root, 0, -1);//root'sfatheris -1
newnode(rs(root),root,-1);//头尾各插入一个空
build(key_value, 1, n, rs(root));
pushup(rs(root));
pushup(root);
}
//旋转,0为左旋,1为右旋 该部分基本固定
void rota(int x, int kind)
{
int y=pre[x];
pushdown(y);
pushdown(x);//必须先把y的标记向下传递,在传x
ch[y][!kind]=ch[x][kind],pre[ch[x][kind]]=y;
if(pre[y])//??y的父节点不是root
ch[pre[y]][ rs(pre[y])==y ]=x;//只能对这句牛逼的代码说声我屮艸芔茻
pre[x]=pre[y];
ch[x][kind]=y,pre[y]=x;
pushup(y);//维护y结点 x节点的信息不用PushUp吗?可以证明这里除了x节点维护的信息不对外,其他所有点信息都正确
//Push_Up(x);//这个可以不写,详见Crash:运用伸展树解决数列维护问题 论文,但是写了也不会多多少时间消耗
}
void Splay(int r, int goal)//将r调整到goal下面
{
pushdown(r);// 离开之前把懒惰标记的信息传递
while(pre[r]!=goal)
{
if(pre[pre[r]]==goal)
rota(r, ch[pre[r]][0]==r);//r在左子树,右旋
else
{
int y=pre[r];
int kind=ch[pre[y]][0]==y;
if(ch[y][kind]==r)//之字形旋转
{//y在其父节点的左子树上,r在y的右子树上
//或者y在其父节点的右子树,r在y左子树
rota(r, !kind);
rota(r, kind);
}
else //一字型旋转
{
rota(y, kind);//注意这里是y啊
rota(r, kind);
}
}
}
pushup(r);
if(goal==0)root=r;
}
//得到第k个结点编号
int getkth(int r, int k)
{
pushdown(r);
int t=sz[ls(r)]+1;
if(t==k)return r;
if(t>k)return getkth(ls(r),k);//在左子树第k个
else return getkth(rs(r), k-t);//在右子树第k-t个
}
//得到以r为根的第一个结点--最左边的节点编号
int getmin(int r)
{
pushdown(r);
while(ls(r))
{
r=ls(r);
pushdown(r);
}
return r;
}
//得到以r为根的最后一个结点--最右边的编号
int getmax(int r)
{
pushdown(r);
while(rs(r))
{
r=rs(r);
pushdown(r);
}
return r;
}
void addv(int l, int r, int d)
{
Splay(getkth(root,l),0);
Splay(getkth(root,r+2),root);
updateadd(key_value, d);
pushup(rs(root));
pushup(root);
}
ll Querysum(int l, int r)
{
Splay(getkth(root, l),0);//第l个点到根结点
Splay(getkth(root, r+2), root);//第r+2个点到根结点的右孩子
return sum[key_value];
}
int main()
{
//freopen("poj3468.txt","r",stdin);
int q;
while(~scanf("%d%d", &n, &q))
{
Init();
while(q--)
{
char op[30];
int x,y,z;
scanf("%s",op);
if(op[0]=='Q')
{
scanf("%d%d",&x,&y);
printf("%lld\n",Querysum(x,y));
}
else
{
scanf("%d%d%d",&x,&y,&z);
addv(x,y,z);
}
}
}
return 0;
}