题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=1540
题目大意:给出n个村庄,m次操作,操作分为三种:D,Q,R
D:炸毁pos位置的村庄。Q查询pos位置的村庄最多的联系村庄。R修好最近一个被炸毁的村庄
思路:线段树区间合并入门题。对于炸毁和修复的,将该点更新状态,维护区间左连续,右连续和中间连续合即可。
ACCode:
#include<stdlib.h>
#include<string.h>
#include<stdio.h>
#include<time.h>
#include<math.h>
// srand(unsigned)time(NULL));rand();
#include<map>
#include<set>
#include<deque>
#include<queue>
#include<stack>
#include<bitset>
#include<string>
#include<fstream>
#include<iostream>
#include<algorithm>
#define ll long long
#define Pii pair<int,int>
#define clean(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MAXN=5e4+10;
const int INF32=0x3f3f3f3f;
const ll INF64=0x3f3f3f3f3f3f3f3f;
const ll MOD=1e9+7;
const double PI=acos(-1.0);
const double EPS=1.0e-8;
//unsigned register
// ios::sync_with_stdio(false)
struct SegTree{
struct Node{
int l,r;
int ls,rs,ms,len;
// int sp,ep;
// int lazy;
};
Node Tree[MAXN<<2];
void PushUp(int rt){
Tree[rt].ms=max(max(Tree[rt<<1].ms,Tree[rt<<1|1].ms),Tree[rt<<1].rs+Tree[rt<<1|1].ls);
Tree[rt].ls=Tree[rt<<1].ls;Tree[rt].rs=Tree[rt<<1|1].rs;
if(Tree[rt<<1].ls==Tree[rt<<1].len) Tree[rt].ls+=Tree[rt<<1|1].ls;
if(Tree[rt<<1|1].rs==Tree[rt<<1|1].len) Tree[rt].rs+=Tree[rt<<1].rs;
}
void Build(int l,int r,int rt){
Tree[rt].l=l;Tree[rt].r=r;
Tree[rt].ls=Tree[rt].rs=Tree[rt].ms=Tree[rt].len=r-l+1;
// Tree[rt].sp=Tree[rt].ep=0;
if(l==r) return ;
int mid=(l+r)>>1;
Build(l,mid,rt<<1);Build(mid+1,r,rt<<1|1);
}
void Update(int Pos,int val,int rt){
if(Tree[rt].l==Tree[rt].r){
Tree[rt].ls=Tree[rt].rs=Tree[rt].ms=val;//该点被炸毁了|| 修好了
return ;
}
if(Pos<=Tree[rt<<1].r) Update(Pos,val,rt<<1);
if(Pos>=Tree[rt<<1|1].l) Update(Pos,val,rt<<1|1);
PushUp(rt);
}
int Query(int Pos,int rt){//在左边
if(Tree[rt].l==Tree[rt].r||Tree[rt].ms==Tree[rt].len||Tree[rt].ms==0) return Tree[rt].ms;
if(Pos<=Tree[rt<<1].r){
if(Pos>=Tree[rt<<1].r-Tree[rt<<1].rs+1) return Query(Pos,rt<<1)+Query(Tree[rt<<1|1].l,rt<<1|1);
else return Query(Pos,rt<<1);
}
if(Pos>=Tree[rt<<1|1].l){//在右边
if(Pos<=Tree[rt<<1|1].l+Tree[rt<<1|1].ls-1) return Query(Pos,rt<<1|1)+Query(Tree[rt<<1].r,rt<<1);
else return Query(Pos,rt<<1|1);
}
}
void Show(int rt){
printf("l=%d r=%d ls=%d rs=%d\n",Tree[rt].l,Tree[rt].r,Tree[rt].ls,Tree[rt].rs);
if(Tree[rt].l==Tree[rt].r) return ;
Show(rt<<1);Show(rt<<1|1);
}
};
SegTree Seg;
int Stk[MAXN],top;
int n,m;
int main(){
while(~scanf("%d%d",&n,&m)){
Seg.Build(1,n,1);top=0;
while(m--){
char opt[5];scanf("%s",&opt);
if(opt[0]=='D'){//删掉x
int x;scanf("%d",&x);
Stk[++top]=x;
Seg.Update(x,0,1);
}
else if(opt[0]=='Q'){// 查询x
int x;scanf("%d",&x);
printf("%d\n",Seg.Query(x,1));
}
else{//修复x
int x=Stk[top--];
Seg.Update(x,1,1);
}//Seg.Show(1);
}
}
}