题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1540
可能是因为hdu这题的数据造的比较水。。本来以为会TLE的结果一次过了,看来map容器的find函数效率也不是那么低嘛
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<algorithm>
#include<map>
using namespace std;
#define MAXN 50050
int n;
int main()
{
int i, j, m, num, ans, dam[MAXN];
map<int, int> tunnel;
map<int, int>::iterator pos;
char op;
memset(dam, 0, MAXN * sizeof(int));
scanf("%d%d", &n, &m);
tunnel.clear();
ans = 0;
j = 0;
for (i = 1; i <= m; i++)
{
scanf("\n");
scanf("%c", &op);
switch (op) {
case 'D':
scanf("%d", &num);
tunnel.insert(make_pair(num, num));//将摧毁的村庄插入map
dam[++j] = num;//单开一个数组记录村庄摧毁的顺序
break;
case 'R':
if (j > 0)
{
num = dam[j--];
pos = tunnel.find(num);//若有被摧毁的村庄,则从map中删除它
tunnel.erase(pos);
}
break;
case 'Q':
scanf("%d", &num);
pos = tunnel.find(num);
if ((pos != tunnel.end()) && (pos->first == num))//若当前村庄被摧毁则一个都不连着,直接输出0
{
printf("0\n");
break;
}
tunnel.insert(make_pair(num, num));
pos = tunnel.find(num);//否则将当前村庄插入map,分别向前向后搜索
ans = n;
if ((pos) != tunnel.begin())
{
pos--;
ans -= pos->first;//减去前方的
pos++;
}
if ((++pos) != tunnel.end())
{
ans -= (n - pos->first + 1);//减去后方的
}
printf("%d\n", ans);//输出最终结果
tunnel.erase(num);
break;
}
}
return 0;
}