文艺平衡树(splay模板)

题目描述

这是一道模板题。

您需要写一种数据结构（可参考题目标题），来维护一个序列，其中需要提供以下操作：

翻转一个区间，例如原有序序列是 5 4 3 2 1，翻转区间是 [2,4] 的话，结果是 5 2 3 4 1。

输入格式

第一行为 n,m，n 表示初始序列有 n 个数，这个序列依次是 1,2,…n−1,n，m 表示翻转操作次数。
接下来 m 行每行两个数 [l,r]，数据保证 1≤l≤r≤n。

输出格式

输出一行 n 个数字，表示原始序列经过 m 次变换后的结果。

样例
Input

5 3
1 3
1 3
1 4

Output

4 3 2 1 5

数据范围与提示
1≤n,m≤105

思路：splay 模板，要理解好splay的原理（感觉有点玄学）
详情见代码

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1e5 + 10;

int n, m;
struct node
{
	int s[2], p, v; // p 为父亲节点编号， v为关键字值，本题为序号
	int size, flag; // size以它为根的子树的大小， flag 是否需要翻转这颗子树， 懒标记

	void init(int _v, int _p) // 初始化
	{
		v = _v, p = _p;
		size = 1;
	}
}tr[N];
int root, idx; // root根节点， idx 加的点的编号

void pushup(int x) // 当旋转或者插入后更新子树的大小
{
	tr[x].size = tr[tr[x].s[0]].size + tr[tr[x].s[1]].size + 1;
}

void pushdown(int x) // 将翻转影响传递下去
{
	if(tr[x].flag)
	{
		swap(tr[x].s[0], tr[x].s[1]);
		tr[tr[x].s[0]].flag ^= 1;
		tr[tr[x].s[1]].flag ^= 1;
		tr[x].flag = 0;
	}
}

void rotate(int x) // 旋转将x旋转到它父亲的位置(通过k，将左右旋转合并了)
{
	int y = tr[x].p, z = tr[y].p; // p为父亲节点编号， y为x的父亲， z为y的父亲
	int k = tr[y].s[1] == x;// k = 0 为左儿子， 1为右儿子
	tr[z].s[tr[z].s[1] == y] = x, tr[x].p = z;// 将y所在的位置换成x, 并将x的父亲换为z
	tr[y].s[k] = tr[x].s[k ^ 1]; // 将原本x在y上的位置换为左/右儿子
	tr[tr[x].s[k ^ 1]].p = y;//将x的左/右儿子的父亲换为y
	tr[x].s[k ^ 1] = y; // x变成y的父亲
	tr[y].p = x;
	pushup(y);//更新
	pushup(x);
}

void splay(int x, int k) // 将x旋转为k的儿子
{
	while(tr[x].p != k)
	{
		int y = tr[x].p, z = tr[y].p; // y是x的父亲， z是y的父亲
		if(z != k)
			if((tr[y].s[1] == x) ^ (tr[z].s[1] == y)) rotate(x); //如果不是一条斜线形状就先旋转x
			else rotate(y);
		rotate(x);
	}
	if(!k) root = x;
}

void insert(int v) // 本题省去建树，直接插入
{
	int u = root, p = 0;
	while(u) p = u, u = tr[u].s[v > tr[u].v]; // 找到下下标v在树中对应的位置
	u = ++ idx; // 开点
	if(p) tr[p].s[v > tr[p].v] = u;
	tr[u].init(v, p);
	splay(u, 0);
}

int get_k(int k) // 得到第k - 1个数的下标
{
	int u = root;
	while(true)
	{
		pushdown(u); // 必须要先看能否翻转
		if(tr[tr[u].s[0]].size >= k) u = tr[u].s[0];
		else if(tr[tr[u].s[0]].size + 1 == k) return u;
		else k -= tr[tr[u].s[0]].size + 1, u = tr[u].s[1];
	}
	return -1;
}

void output(int u)
{
	pushdown(u);
	if(tr[u].s[0]) output(tr[u].s[0]);
	if(tr[u].v >= 1 && tr[u].v <= n) printf("%d ", tr[u].v);
	if(tr[u].s[1]) output(tr[u].s[1]);
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 0; i <= n + 1; i++) insert(i);
	int x = root;
	while(m --)
	{
		int l, r;
		scanf("%d%d", &l, &r);
		l = get_k(l), r = get_k(r + 2); // 跟新pushdown
		//目的是在该次splay中所有用到的点不需要pushdown了
		cout << "****  " << l << " " << r << "   *********" << endl;
		splay(l, 0); //将l提到根
		splay(r, l); //将 r的父亲变成l
		tr[tr[r].s[0]].flag ^= 1; // 翻转区间
		//output(root); cout << endl;
	}
	output(root);
	return 0;
}