[codevs 1343] 蚱蜢(省队选拔赛湖南)

http://codevs.cn/problem/1343/

题解:

本题splay基本操作:

1.如果是左跳,比如从 x 左跳到 y,就相当于查询 [ y, x ) 区间的最大值,那么就把 y-1 伸展到根,把 x 伸展到根的右节点,那么根的右节点的左节点对应的就是这段区间。

1.如果是右跳,比如从 x 右跳到 y,就相当于查询 ( x, y ] 区间的最大值,那么就把 x 伸展到根,把 y+1 伸展到根的右节点,那么根的右节点的左节点对应的就是这段区间。

3.这里 splay 是要将某个位置的节点伸展到给定位置,所以采用从顶向下查询第 rank 个的方式伸展,见代码。

4.remove操作:删除第 k 个,就把第 k-1 个伸展到根,把第 k+1 个伸展到根的右节点,删除它的左节点即可。别忘记 update 操作。

5.insert操作:如果要插到第 k 处,就把第 k-1 个节点旋转到根,把第 k 个节点旋转到根的右节点,那么插入到根的右节点的左节点就可以代替原来的第 k 个。好像没必要像我分两种情况。最后别忘 update,先子节点再根节点。

6.move操作:就是先 remove 再 insert。

7.宏定义:因为用到大量的根节点的左右节点的调用,而这里还不适合用引用,所以干脆定义了宏,简单清楚,编程效果好了很多。

8.边界防溢出:因为1、2中的将 y-1、y+1 伸展到根都是危险操作,因为 y 有可能等于 1 或 n,而 0 节点不能成为根节点,因为0节点默认为空;n+1 节点直接溢出,所以建立两个虚拟节点,一个为1,一个为 n+1,分别放在开头和末尾,对编程本身没影响,但不用担心溢出了,不过各个节点相应的位置就变了,读入位置 x 后 x++ 就行了。


代码:

总时间耗费: 6439ms 
总内存耗费: 3 kB

有些慢,但内存耗的很少。

#include<cstdio>
#include<algorithm>
using namespace std;

const int INF = 1e9 + 7;
const int maxn = 100000 + 10;

int root, ch[maxn][2], v[maxn], s[maxn], maxv[maxn];

void update(int o) {
	maxv[o] = max(v[o], maxv[ch[o][0]]);
	maxv[o] = max(maxv[o], maxv[ch[o][1]]);
	s[o] = s[ch[o][0]] + s[ch[o][1]] + 1;
}

int cmp(int o, int k) {
	int sum = s[ch[o][0]] + 1;
	if(sum == k) return -1;
	return k < sum ? 0 : 1;
}

void rotate(int& o, int d) {
	int k = ch[o][d^1]; ch[o][d^1] = ch[k][d]; ch[k][d] = o;
	update(o); update(k); o = k;
}

void splay(int& o, int k) {
	int d = cmp(o, k);
	if(d == -1) return;
	if(d == 1) k -= s[ch[o][0]] + 1;
	int p = ch[o][d];
	int d2 = cmp(p, k);
	int k2 = (d2 == 0) ? k : k - s[ch[p][0]] - 1;
	if(d2 != -1) {
		splay(ch[p][d2], k2);
		if(d2 == d) rotate(o, d^1); else rotate(ch[o][d], d);
	}
	rotate(o, d^1);
}

void build(int l, int r, int pa) {
	if(l > r) return;
	if(l == r) { 
		maxv[l] = v[l]; s[l] = 1; 
		if(l < pa) ch[pa][0] = l; else ch[pa][1] = l;
		return;
	} 
	int m = (l+r) >> 1; maxv[m] = v[m];
	build(l, m-1, m); build(m+1, r, m); update(m);
	if(m < pa) ch[pa][0] = m; else ch[pa][1] = m;
}

#define lc ch[root][0]
#define rc ch[root][1]

void move(int x, int y) {
	splay(root, x-1); splay(rc, x-s[lc]); //x+1 - (s[lc]+1)
	int o = ch[rc][0];
	ch[rc][0] = 0; //remove
	update(rc);
	if(x < y) {
		splay(root, y-1); 
		splay(rc, y-s[lc]-1); //y - (s[lc]+1)
		ch[rc][0] = o; update(rc); update(root); //notice
	} else {
		splay(root, y); 
		splay(lc, y-1); //
		ch[lc][1] = o; update(lc); update(root);
	}
}

int main() {
	int n, m; 
	scanf("%d%d", &n, &m); 
	for(int i = 2; i <= n+1; i++) scanf("%d", &v[i]); 
	build(1, n+2, 0); 
	root = (n+3) >> 1;
	
	for(int i = 1; i <= m; i++) {
		int x, k; 
		char cmd[1]; 
		scanf("%d%s%d", &x, &cmd, &k); 
		x++;
		switch(cmd[0]) {
			case 'L': {
				splay(root, x-k-1); 
				splay(rc, x-s[lc]-1); //x - (s[lc]+1)
				printf("%d\n", maxv[ch[rc][0]]); 
				move(x, x-k);
				break; 
			}
			case 'D': {
				splay(root, x); 
				splay(rc, x+k-s[lc]); //x+k+1 - (s[lc]+1)
				printf("%d\n", maxv[ch[rc][0]]); 
				move(x, x+k);
				break;
			}
		}
	}
	
	return 0; 
}


posted @ 2015-02-05 22:55  wfwbz  阅读(167)  评论(0编辑  收藏  举报