洛谷 P1503 鬼子进村

题目背景

小卡正在新家的客厅中看电视。电视里正在播放放了千八百次依旧重播的《亮剑》，剧中李云龙带领的独立团在一个县城遇到了一个鬼子小队，于是独立团与鬼子展开游击战。

题目描述

描述 县城里有n个用地道相连的房子，第i个只与第i-1和第i+1个相连。这是有m个消息依次传来

1、消息为D x：鬼子将x号房子摧毁了，地道被堵上。

2、消息为R ：村民们将鬼子上一个摧毁的房子修复了。

3、消息为Q x：有一名士兵被围堵在x号房子中。

李云龙收到信息很紧张，他想知道每一个被围堵的士兵能够到达的房子有几个。

输入输出格式

输入格式：

第一行2个整数n，m（n，m<=50000）。

接下来m行，有如题目所说的三种信息共m条。

输出格式：

对于每一个被围堵的士兵，输出该士兵能够到达的房子数。

输入输出样例

输入样例#1：

7 9
D 3
D 6
D 5
Q 4
Q 5
R
Q 4
R
Q 4

输出样例#1： ·

1
0
2
4

说明

若士兵被围堵在摧毁了的房子中，那只能等死了。。。。。。

思路：对于第一种操作，我们可以开一个vis数组记录某个地洞是否被堵了，同时要把堵了的地洞加入一个栈，为后面的第二种操作做准备，第二种操作，直接就把栈顶元素弹出来就好了，再把相应的vis数组清空，但是，第三种操作，貌似有些棘手，但是，我们仔细想一下，被困队员能到达的地洞可以认为是一个线性的数列，也就是说，是有左右边界的，而左右边界又正好是这个被困队员所在的被堵洞口的前驱和后继！用splay来维护一下就可以解决了！

代码：

#include<cstdio>
#include<cctype>
#include<stack>
#define maxn 50007
using namespace std;
int n,m,root,tot;
char c[2];
inline int qread() {
  char c=getchar();int num=0,f=1;
  for(;!isdigit(c);c=getchar()) if(c=='-') f=-1;
  for(;isdigit(c);c=getchar()) num=num*10+c-'0';
  return num*f;
}
bool vis[maxn];
struct Tree {
  int fa,ch[2],val;
}t[maxn];
inline void rotate(int x) {
  int y=t[x].fa,z=t[y].fa;
  int k=t[y].ch[1]==x;
  t[z].ch[t[z].ch[1]==y]=x;
  t[x].fa=z;
  t[y].ch[k]=t[x].ch[k^1];
  t[t[x].ch[k^1]].fa=y;
  t[x].ch[k^1]=y,t[y].fa=x;
}
inline void splay(int x, int rt) {
  while(t[x].fa!=rt) {
  	int y=t[x].fa,z=t[y].fa;
  	if(z!=rt) (t[y].ch[0]==x)^(t[z].ch[0]==y)?rotate(x):rotate(y);
  	rotate(x);
  }
  if(!rt) root=x;
}
inline void insert(int x) {
  int u=root,f=0;
  while(u&&t[u].val!=x) {
  	f=u;
  	u=t[u].ch[x>t[u].val];
  }
  if(u);
  else {
  	u=++tot;
  	if(f) t[f].ch[x>t[f].val]=u;
  	t[u].fa=f,t[u].val=x;
  }
  splay(u,0);
}
inline void find(int x) {
  int u=root;
  if(!u) return;
  while(t[u].ch[x>t[u].val]&&x!=t[u].val) u=t[u].ch[x>t[u].val];
  splay(u,0);
}
inline int nxt(int x, int f) {
  find(x);
  int u=root;
  if(t[u].val>x&&f) return u;
  if(t[u].val<x&&!f) return u;
  u=t[u].ch[f];
  while(t[u].ch[f^1]) u=t[u].ch[f^1];
  return u;
}
inline void sc(int x) {
  int last=nxt(x,0),next=nxt(x,1);
  splay(last,0),splay(next,last);
  t[next].ch[0]=0;
}
stack<int>q;
int main() {
  n=qread(),m=qread();
  insert(0),insert(n+1);
  for(int i=1,x;i<=m;++i) {
  	scanf("%s",c);
  	if(c[0]=='D') {
  	  x=qread(),vis[x]=1;
	  q.push(x),insert(x);
	}
	if(c[0]=='R') {
	  vis[q.top()]=0;
	  sc(q.top());
	  q.pop();
	}
	if(c[0]=='Q') {
	  x=qread();
	  if(vis[x]) printf("0\n");
	  else printf("%d\n",t[nxt(x,1)].val-t[nxt(x,0)].val-1);
	}
  }
  return 0;
}