【bzoj1269】【AHOI2006】文本编辑器editor【Splay】

最新推荐文章于 2019-03-07 15:35:00 发布

原创最新推荐文章于 2019-03-07 15:35:00 发布 · 757 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Splay #bzoj #文本编辑 #c++

bzoj 同时被 2 个专栏收录

22 篇文章

订阅专栏

平衡树

2 篇文章

订阅专栏

本文讨论了使用平衡树解决字符串插入问题时遇到的挑战，并提供了有效的解决方案。通过详细解释平衡树的操作，如翻转标记、节点旋转等，作者成功地解决了字符串插入过程中的空字符跳过问题。此外，文章还介绍了如何使用特定函数实现字符串的建树和树根挂接，以及如何通过调整代码结构避免低级错误。通过实例代码演示，读者可以更好地理解并应用这些技巧。

被文艺平衡树折磨了一天以后发现这道题就很好做啦~~~

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1269
但是c++喜闻乐见的gets十分不好使。

如果insert的字符串有空格，gets会跳过去不读空格。

好在题目给了ASCII码的范围。

这样就可做了。
写的时候不要把功能一次性全写完，先写个基本的，排除一下低级错误……我把Move和insert写完以后发现Splay又出了几个沙茶错误- -

还有居然把Next和Prev搞反了- -……

Splay中cur代表当前光标位置。

flip是标记是否反转的标记~

p表示当前节点是他爸的左孩子还是右孩子。

build是用来给字符串建树，建完之后把这棵树的树根直接挂到上面去~（预先把光标两侧的节点Splay到一块去）。

代码在下面。

#include<cstring>
#include<iostream>
#include<cstdio>
using namespace std;
char str[3000002],cmd[10];
int n,x;
struct Splay{
	static const int maxn=3000001;
	struct node{
		int ch[2],f,s;
		char v;
		bool p,flip;
	}t[maxn];
	int rt,tot,cur;
	inline void upd(int x){
		t[x].s=t[t[x].ch[0]].s+t[t[x].ch[1]].s+1;
	}
	inline void pushdown(int x){
		if(!t[x].flip) return;
		t[t[x].ch[0]].flip^=1;
		t[t[x].ch[0]].p^=1;
		t[t[x].ch[1]].flip^=1;
		t[t[x].ch[1]].p^=1;
		swap(t[x].ch[0],t[x].ch[1]);
		t[x].flip=0;
	}
	inline void sc(int a,int b,bool c){//把节点a的c孩子置为b（c=0：左孩子，c=1：右孩子）
		t[a].ch[c]=b;
		t[b].f=a;
		t[b].p=c;
	}
	inline void rot(int x){
		int y=t[x].f,p=t[x].p;
		if(y==rt) t[rt=x].f=0;
		else sc(t[y].f,x,t[y].p);
		sc(y,t[x].ch[!p],p);
		sc(x,y,!p);
		upd(y);
	}
	void splay(int x,int r){
		int f;
		while((f=t[x].f)!=r){
			if(t[f].f==r) rot(x);
			else if(t[f].p==t[x].p) rot(f),rot(x);
			else rot(x),rot(x);
//			cout<<x<<"\n";
		}
		upd(x);
	}
	int build(int l,int r,int f,int p){
		if(l>r) return 0;
		int now=++tot;
		t[now].f=f;
		t[now].p=p;
		t[now].flip=false;
		t[now].v=str[l+r>>1];
		t[now].ch[0]=build(l,(l+r>>1)-1,now,0);
		t[now].ch[1]=build((l+r>>1)+1,r,now,1);
		upd(now);
		return now;
	}
	int select(int x,int k){
		while(1){
			pushdown(x);
			if(k==t[t[x].ch[0]].s+1)return x;
			else if(k<t[t[x].ch[0]].s+1) x=t[x].ch[0];
			else k-=t[t[x].ch[0]].s+1,x=t[x].ch[1];
		}
	}
	inline int insert(int subtree){
		splay(select(rt,cur+1),0);
		splay(select(rt,cur+2),rt);
		sc(t[rt].ch[1],subtree,0);
		upd(t[rt].ch[1]);
		upd(rt);
	}
	inline int Flip(int length){
		splay(select(rt,cur+1),0);
		splay(select(rt,cur+2+length),rt);
		t[t[t[rt].ch[1]].ch[0]].flip^=1;
	}
	inline void del(int length){
		splay(select(rt,cur+1),0);
		splay(select(rt,cur+2+length),rt);
		sc(t[rt].ch[1],0,0);
		upd(t[rt].ch[1]);
		upd(rt);
	}
	inline char get(){
		return t[select(rt,cur+2)].v;
	}
	Splay():rt(0),tot(0),cur(0){}
}editor;
int main(){
	scanf("%d\n",&n);
	editor.rt=editor.build(0,1,0,0);
	while(n--){
		scanf("%s",cmd);
		switch(cmd[0]){
			case 'M':
				scanf("%d",&x);
				editor.cur=x;
				break;
			case 'I':
				scanf("%d",&x);
				do{str[0]=getchar();}while(str[0]<32||str[0]>126);
				for(int i=1;i<x;++i) str[i]=getchar();
				getchar();
				editor.insert(editor.build(0,x-1,0,0));
				break;
			case 'D':
				scanf("%d",&x);
				editor.del(x);
				break;
			case 'R':
				scanf("%d",&x);
				editor.Flip(x);
				break;
			case 'G':
				putchar(editor.get());
				putchar('\n');
				break;
			case 'P':
				editor.cur--;
				break;
			case 'N':
				editor.cur++;
				break;
		}
	}
	return 0;
}