男儿何不带吴钩,收取关山五十州。
征战天下是秋实大哥一生的梦想,所以今天他又在练习一个对战游戏。
秋实大哥命令所有士兵从左到右排成了一行来抵挡敌人的攻击。
敌方每一次会攻击一个士兵,这个士兵就会阵亡,整个阵列就会从这个位置断开;同时有的时候已阵亡的士兵会受人赢气息感染而复活。
秋实大哥想知道某一时刻某一个士兵所在的阵列的长度是多少。
Input
第一行包含两个整数n,m,表示秋实大哥的士兵数目和接下来发生的事件数目。
接下来m行,每一行是以下三种事件之一:
0 x : 表示x位置的士兵受到攻击阵亡
1 x : 表示x位置的士兵受人赢气息感染复活
2 x : 秋实大哥想知道第x个士兵所在阵列的长度
(1≤n,m≤100000,1≤x≤n。)
Output
对于每一个2 x事件,输出对应的答案占一行。
样例输入
5 3
2 2
0 3
2 2
样例输出
5
2
题解:平衡树,每死亡一个士兵,就将其编号加到平衡树中;如果当前找寻的士兵死亡,那么就输出0,否则直接在平衡树中找和这个士兵的编号最近的两个已经死亡的士兵的编号,然后就得出答案了。(如果有复活的话就直接把复活的编号从平衡树中删掉就好啦。)
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<set>
#include<cstdlib>
using namespace std;
set<int>tr;
int n,m,l,r;
bool vis[100004];
int main(){
scanf("%d%d",&n,&m);
tr.clear();
tr.insert(0);tr.insert(n+1);
for(int i=1;i<=m;i++){
int t,x;
scanf("%d%d",&t,&x);
if(t == 0){
if(!vis[x])vis[x]=1,tr.insert(x);
}
if(t == 1){
if(vis[x]) vis[x]=0,tr.erase(x);
}
if(t == 2){
if(vis[x]) puts("0");
else {
l = *--tr.lower_bound(x);
r = *tr.upper_bound(x);
printf("%d\n",r-l-1);
}
}
}
return 0;
}