hdu1540 Tunnel Warfare--单点更新查询 & 区间合并

原题链接： ​​http://acm.hdu.edu.cn/showproblem.php?pid=1540​​

题意：

一条直线，n个点，标号1-----n，m个操作，规则如下：

D x ：摧毁点x

Q x  ：查询包含点x的最长序列

R ： 恢复上一次被摧毁的一个点

思路就是：我们把所有点初始为1，摧毁该点就赋值为0，用栈保存摧毁的点，当需要恢复时就取出栈顶，进行更新。

#define _CRT_SECURE_NO_DEPRECATE 

#include<iostream>
#include<string.h>
#include<algorithm>
using namespace std;

int s[50005];//模拟栈，把删除的数放进去，待rebuild的时候取出
int top;

struct Node
{
  int l, r, mid;
  int len;
  int lsum, rsum, msum;
}p[50005 * 4];

void pushUp(int n)
{
  p[n].lsum = p[n * 2].lsum + (p[n * 2].lsum == p[n * 2].len ? p[n * 2 + 1].lsum : 0);
  p[n].rsum = p[n * 2 + 1].rsum + (p[n * 2 + 1].rsum == p[n * 2 + 1].len ? p[n * 2].rsum : 0);
  p[n].msum = max(max(p[n * 2].msum, p[n * 2 + 1].msum), p[n * 2].rsum + p[n * 2 + 1].lsum);
}

void build(int n, int l, int r)
{
  p[n].l = l;
  p[n].r = r;
  p[n].mid = (l + r) / 2;
  p[n].len = r - l + 1;
  p[n].lsum = p[n].rsum = p[n].msum = p[n].len;
  if (l == r)
    return;

  build(n * 2, l, p[n].mid);
  build(n * 2 + 1, p[n].mid + 1, r);
}

void update(int n, int pos, int c)
{
  if (p[n].l == p[n].r)
  {
    if (c == 1)
      p[n].lsum = p[n].rsum = p[n].msum = 1;//rebuild
    else
      p[n].lsum = p[n].rsum = p[n].msum = 0;//destroy

    return;//要return，害死人啊
  }

  if (pos <= p[n].mid)
    update(n * 2, pos, c);
  else
    update(n * 2 + 1, pos, c);

  pushUp(n);
}

int query(int n, int pos)
{
  if (p[n].l == p[n].r || p[n].msum == p[n].len || p[n].msum == 0)//到了叶子节点或者该访问区间为空或者已满都不必要往下走了  
    return p[n].msum;

  if (pos <= p[n].mid)
  {
    if (pos >= p[n * 2].r - p[n * 2].rsum + 1)
      return p[n * 2].rsum + p[n * 2 + 1].lsum;
    else
      return query(n * 2, pos);
  }
  else
  {
    if (pos <= p[n * 2 + 1].l + p[n * 2 + 1].lsum - 1)
      return p[n * 2].rsum + p[n * 2 + 1].lsum;
    else
      return query(n * 2 + 1, pos);
  }
}

int main()
{
  char ch[5];
  int n, m;
  int pos;
  while (~scanf("%d%d", &n, &m))
  {
    top = 0;

    build(1, 1, n);

    while (m--)
    {
      scanf("%s", ch);

      if (ch[0] == 'D')
      {
        scanf("%d", &pos);
        s[top++] = pos;
        update(1, pos, 0);//0代表destroy
      }
      else if (ch[0] == 'Q')
      {
        scanf("%d", &pos);
        printf("%d\n", query(1, pos));
      }
      else
      {
        if (top > 0)//栈还有数的话,注意这里要判断
        {
          pos = s[--top];
          update(1, pos, 1);//1代表rebuild
        }
      }
    }
  }

  return 0;
}