P6166 [IOI2012] scrivener

本题与这题基本完全一样，只是字符的下标这里是从 $0$ 开始。

显然，这是可持久化线段树的较为简单的模板题。撤销事实上只是换根，就是将当前的根换成 $x$ 次操作之前的根，线段树动态开点即可。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
using namespace std;

const int N = 1e6 + 5;

int n;

class Persis_Segment_Tree
{
public:
	struct Node
	{
		int l, r;
		char val;
		int cnt;
	}tr[N * 25];
	int root[N], idx;
	Persis_Segment_Tree() : idx(1) {}
	void insert(int& now, int pr, int l, int r, char p)
	{
		now = ++idx;
		tr[now] = tr[pr];
		if (l > r) return;
		if (l == r)
		{
			tr[now].cnt++;
			tr[now].val = p;
			return;
		}
		int mid = l + r >> 1;
		if (tr[tr[pr].l].cnt == mid - l + 1)
		{
			insert(tr[now].r, tr[pr].r, mid + 1, r, p);
		}
		else insert(tr[now].l, tr[pr].l, l, mid, p);
		tr[now].cnt = tr[tr[now].l].cnt + tr[tr[now].r].cnt;
	}
	char query(int u, int l, int r, int k)
	{
		if (l >= r) return tr[u].val;
		int mid = l + r >> 1;
		if (k <= tr[tr[u].l].cnt) return query(tr[u].l, l, mid, k);
		return query(tr[u].r, mid + 1, r, k - tr[tr[u].l].cnt);
	}
};

Persis_Segment_Tree sg;

int main()
{
	scanf("%d", &n);
	int cnt = 0;
	for (int i = 1; i <= n; i++)
	{
		char a, b;
		cin >> a;
		if (a == 'T')
		{
			cin >> b;
			cnt++;
			sg.insert(sg.root[cnt], sg.root[cnt - 1], 1, n, b);
		}
		else if (a == 'U')
		{
			int x;
			scanf("%d", &x);
			cnt++;
			sg.root[cnt] = sg.root[cnt - x - 1];
		}
		else
		{
			int x;
			scanf("%d", &x);
			x++;
			cout << sg.query(sg.root[cnt], 1, n, x) << "\n";
		}
	}
	return 0;
}