题目链接:HDU - 1540
题意:给出数字1~n和三种操作
操作D:删除一个数(但是那个位置依然空在那)。
操作Q:询问一个数,回答包括这个数在内的连续区间的长度。(若那个数被删,返回0)
操作R:重建最后被删除的数。
题解:对于询问Q,如果我们每次都遍历1~n寻找连续区间的长度,肯定是要超时的。这题可以用线段树维护求解。但我
这里给出另一种解法。每一个Q我们只要知道所询问的数所在的连续区间的左端点和右端点然后相减即可。那我
们如何快速知道端点呢?因为数字1~n是有序的,删除的数也是有序的,那么将D删除的数放入set中,再利用二
分查找Q询问的数在set中所在的区间,这样就可以快速求出端点了。至于R重建,用个栈记录一下即可。
#include <bits/stdc++.h>
using namespace std;
int main()
{
int n, m, x;
char c;
while(cin>>n>>m){
set<int> se;
se.insert(0); //开始的没有删除数,所有数的上下界都是0和n+1
se.insert(n + 1);
stack<int> st;
while(m--){
cin>>c;
if(c == 'D'){
cin>>x;
se.insert(x); //记录删除的数
st.push(x); //记录删除的顺序
}
else if(c == 'Q'){
cin>>x;
if(se.count(x)) cout<<0<<endl; //判断x是否被删除
else{
set<int>::iterator it = se.lower_bound(x); //因为x不在set中,二分查的一定是最小上界
int r = *it; //右端点
int l = *(--it); //左端点
cout<<r - l - 1<<endl;
}
}
else{
x = st.top(); st.pop(); //重建栈顶元素
se.erase(se.find(x)); //在set中删除即为重建
}
}
}
return 0;
}