Tunnel Warfare
题目链接:
http://acm.hdu.edu.cn/showproblem.php?pid=1540
解题思路:
D代表破坏村庄,R代表修复最后被破坏的那个村庄,Q代表询问包括x在内的最大连续区间是多少。
用线段树来记录。然后用三个变量分别代表左边连续区间,右边连续区间和最大连续区间。
AC代码:
#include <iostream>
#include <cstdio>
#include <stack>
#include <algorithm>
using namespace std;
const int maxn = 50005;
int n,m;
stack<int> sta;
struct node{
int l,r;
int ls,rs,ms;
//ls,左端最大连续区间,rs右端最大连续区间,ms区间内最大连续区间
}tree[maxn<<2];
void build(int id,int l,int r){
tree[id].l = l;
tree[id].r = r;
tree[id].ls = tree[id].rs = tree[id].ms = r-l+1;
if(l == r)
return ;
int mid = (l+r)>>1;
build(id<<1,l,mid);
build(id<<1|1,mid+1,r);
}
void update(int id,int x,int val){
if(tree[id].l == tree[id].r){
if(val == 1)
tree[id].ls = tree[id].rs = tree[id].ms = 1;//修复
else
tree[id].ls = tree[id].rs = tree[id].ms = 0;//破坏
return ;
}
int mid = (tree[id].l+tree[id].r)>>1;
if(x <= mid)
update(id<<1,x,val);
else
update(id<<1|1,x,val);
tree[id].ls = tree[id<<1].ls;//左区间
tree[id].rs = tree[id<<1|1].rs;//右区间
tree[id].ms = max(max(tree[id<<1].ms,tree[id<<1|1].ms),tree[id<<1].rs+tree[id<<1|1].ls);
if(tree[id<<1].ls == tree[id<<1].r-tree[id<<1].l+1)//左子树区间如果满了,则父亲左区间要加上右孩子的左区间
tree[id].ls += tree[id<<1|1].ls;
if(tree[id<<1|1].rs == tree[id<<1|1].r-tree[id<<1|1].l+1)//同理
tree[id].rs += tree[id<<1].rs;
}
int query(int id,int x){
if(tree[id].l == tree[id].r ||tree[id].ms == tree[id].r-tree[id].l+1)//到了叶子节点或者该访问区间为空或者已满都不必要往下走了
return tree[id].ms;
int mid = (tree[id].l+tree[id].r)>>1;
if(x <= mid){
if(x >= tree[id<<1].r-tree[id<<1].rs+1)//因为t<=mid,看左子树,a[2*i].r-a[2*i].rs+1代表左子树右边连续区间的左边界值,如果t在左子树的右区间内,则要看右子树的左区间有多长并返回
return query(id<<1,x)+query(id<<1|1,mid+1);
else
return query(id<<1,x);//如果不在左子树的右边界区间内,则只需要看左子树
}
else{
if(x <= tree[id<<1|1].l+tree[id<<1|1].ls-1)//同理
return query(id<<1|1,x)+query(id<<1,mid);
else
return query(id<<1|1,x);
}
}
int main(){
while(~scanf("%d%d",&n,&m)){
while(!sta.empty())
sta.pop();
char op[5];
int x;
build(1,1,n);
while(m--){
scanf("%s",op);
if(op[0] == 'D'){
scanf("%d",&x);
sta.push(x);
update(1,x,0);
}
else if(op[0] == 'Q'){
scanf("%d",&x);
printf("%d\n",query(1,x));
}
else{
if(!sta.empty()){
x = sta.top();
sta.pop();
update(1,x,1);
}
}
}
}
return 0;
}