[UESTC 1061]秋实大哥与战争

男儿何不带吴钩,收取关山五十州。
征战天下是秋实大哥一生的梦想,所以今天他又在练习一个对战游戏。
秋实大哥命令所有士兵从左到右排成了一行来抵挡敌人的攻击。
敌方每一次会攻击一个士兵,这个士兵就会阵亡,整个阵列就会从这个位置断开;同时有的时候已阵亡的士兵会受人赢气息感染而复活。
秋实大哥想知道某一时刻某一个士兵所在的阵列的长度是多少。
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;
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值