【JZOJ3794】【洛谷P1383】高级打字机【主席树】

最新推荐文章于 2022-07-25 06:53:00 发布

stoorz1023

最新推荐文章于 2022-07-25 06:53:00 发布

阅读量210

点赞数

分类专栏：主席树

本文链接：https://blog.csdn.net/SSL_ZYC/article/details/88750496

版权

主席树专栏收录该内容

9 篇文章 0 订阅

订阅专栏

题目大意：

题目链接：

洛谷：https://www.luogu.org/problemnew/show/P1383
JZOJ：https://jzoj.net/senior/#main/show/3794

要求一种数据结构满足这样的操作：

$T\ x$ ：在文章末尾打下一个小写字母 $x$ 。
$U\ x$ ：撤销最后的 $x$ 次修改操作。
$Q\ x$ ：询问当前文章中第 $x$ 个字母并输出。

思路：

吐槽

我是不是可以去IOI了XD

这道题很显然是用主席树维护这个序列。干脆直接开一个长度为 $10^5$ 的主席树。每一个叶子节点储存这个节点的字母。这样虽然会大大增加空间但是不会MLE啊XD
主席树的基础题吧。对于每一个 $T$ 操作，新开一个主席树。对于每一个 $U$ 操作，把第 $c n t - x - 1$ 个主席树抠到第 $c n t$ 个主席树中。对于每一个 $Q$ 操作，直接输出第 $c n t$ 个主席树的第 $x$ 位即可。
时间复杂度 $O(T\ log\ n)$ ，空间复杂度 $\ log\ n)$

代码：

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

const int N=100010;
int T,tot,sum[N],cnt,x,root[N];
char ch;

struct Tree
{
	int ls,rs;
	char c;
}tree[N*20];

int build(int l,int r)
{
	int p=++tot;
	if (l<r)
	{
		int mid=(l+r)/2;
		tree[p].ls=build(l,mid);
		tree[p].rs=build(mid+1,r);
	}
	return p;
}

int add(int now,int l,int r,int k,char val)  //插入
{
	int p=++tot;
	tree[p]=tree[now];
	if (l==r) tree[p].c=val;
	else
	{
		int mid=(l+r)/2;
		if (k<=mid) tree[p].ls=add(tree[now].ls,l,mid,k,val);
			else tree[p].rs=add(tree[now].rs,mid+1,r,k,val);
	}
	return p;
}

char ask(int x,int l,int r,int k)  //询问
{
	if (l==k&&r==k) return tree[x].c;  //找到
	int mid=(l+r)/2;
	if (k<=mid) return ask(tree[x].ls,l,mid,k);
		else return ask(tree[x].rs,mid+1,r,k);
}

int main()
{
	scanf("%d",&T);
	root[0]=build(1,N);
	while (T--)
	{
		cin>>ch;
		if (ch=='T')
		{
			cin>>ch;
			cnt++;
			sum[cnt]=sum[cnt-1]+1;
			root[cnt]=add(root[cnt-1],1,N,sum[cnt],ch);
		}
		else if (ch=='U')
		{
			scanf("%d",&x);
			cnt++;
			root[cnt]=root[cnt-x-1];
			sum[cnt]=sum[cnt-x-1];
		}
		else
		{
			scanf("%d",&x);
			putchar(ask(root[cnt],1,N,x));
			putchar(10);
		}
	}
	return 0;
}