Tunnel Warfare(树状数组+二分)

http://poj.org/problem?id=2892

题意：输入n,m。n代表数轴的长度，m代表操作数。

D x: 摧毁点x

Q x: 询问村庄x最左与最右没有被摧毁的点的距离

R  ：恢复最后一个被摧毁的点

 #include <stdio.h>
 #include <string.h>
 const int N=;
 int c[N],keep[N],n;
 bool vis[N];

 int lowbit(int x)
 {
     return x&(-x);
 }
 int sum(int x)
 {
     int res = ;
     while(x > )
     {
         res+=c[x];
         x-=lowbit(x);
     }
     return res;
 }
 void add(int x,int d)
 {
     while(x <= n)
     {
         c[x]+=d;
         x+=lowbit(x);
     }
 }
 int binary_find(int key)
 {
     int l = ,r = n+,pos = n+;
     while(l <= r)
     {
         int mid = (l+r)>>;
         if (sum(mid) >= key)
         {
             r = mid-;
             pos = mid;
         }
         else
             l = mid+;
     }
     return pos;
 }
 int main()
 {
     char ch;
     int m,x,t = ;
     scanf("%d%d",&n,&m);
     for (int i = ; i < m; i++)
     {
         getchar();
         scanf("%c",&ch);
         if (ch=='D')
         {
             scanf("%d",&x);
             keep[++t] = ++x;
             vis[x] = true;
             add(x,);
         }
         else if (ch=='R')
         {
             x = keep[t--];
             vis[x] = false;
             add(x,-);
         }
         else
         {
             int pos1,pos2;
             scanf("%d",&x);
             x++;
             if (vis[x])
                 printf("0\n");
             else
             {
                 x = sum(x);
                 pos1 = binary_find(x);
                 pos2 = binary_find(x+);
                 printf("%d\n",pos2-pos1-);
             }
         }
     }
     return ;
 }

STL 做法

 #include <stdio.h>
 #include <set>
 #include <stack>
 using namespace std;

 int main()
 {
     char ch;
     int n,m,x;
     set<int>s;
     stack<int>p;
     scanf("%d%d",&n,&m);
     s.insert();//增加最小边界，保证查找第一个位置时它的前一个位置存在
     s.insert(n+);//增加最大边界，保证查找最后一个位置时它的后一位置存在
     while(m--)
     {
         getchar();
         scanf("%c",&ch);
         if (ch=='D')
         {
             scanf("%d",&x);
             s.insert(x);//s中储存被摧毁的点，且自动将键值从小到大排列
             p.push(x);//用栈保存被摧毁的点，方便最后一个被摧毁的点的恢复
         }
         else if (ch=='R')
         {
             x = p.top();//取出最后一个被摧毁的点
             p.pop();//从栈中删除该点
             s.erase(x);//从s中删除该点
         }
         else
         {
             scanf("%d",&x);
             int r = *(s.lower_bound(x));//r表示值>=x的第一个值，即x右边的被摧毁的点中最接近x的点
             int l = *(--s.lower_bound(x));//l表示x左边的被摧毁的点中最接近x的点
             if (r==x)//如果查询的点在s中被找到，说明该点已被摧毁
                 printf("0\n");
             else
                 printf("%d\n",r-l-);//与x连续的没被摧毁的村庄个数
         }
     }
     return ;
 }