首页 > 代码库 > hdu-1540-Tunnel Warfare-线段树
hdu-1540-Tunnel Warfare-线段树
lmax[i]:i区间内,从左往右,连续的最大长度
rmax[i]:i区间内,从右往左,连续的最大长度
val[i]:i区间的长度。
#include<stdio.h> #include<string.h> #include<iostream> #include<algorithm> #include<math.h> #include<stack> using namespace std; #define INF 99999999 #define lmin 1 #define rmax n #define lson l,(l+r)/2,rt<<1 #define rson (l+r)/2+1,r,rt<<1|1 #define root lmin,rmax,1 #define maxn 55000 int lmax[maxn*4]; int rmax[maxn*4]; int val[maxn*4]; void push_up(int rt) { } void push_down(int rt) { } void creat(int l,int r,int rt) { lmax[rt]=rmax[rt]=val[rt]=r-l+1; if(l!=r) { creat(lson); creat(rson); } } void update(int x,int l,int r,int rt) { if(r<x||l>x)return ; lmax[rt]=min(x-l,lmax[rt]); rmax[rt]=min(rmax[rt],r-x); if(l!=r) { update(x,lson); update(x,rson); } } void update1(int x,int l,int r,int rt) { if(r<x||l>x)return ; if(l!=r) { update1(x,lson); update1(x,rson); lmax[rt]=lmax[rt<<1]; rmax[rt]=rmax[rt<<1|1]; if(lmax[rt<<1]==val[rt<<1]) lmax[rt]=lmax[rt<<1]+lmax[rt<<1|1]; if(rmax[rt<<1|1]==val[rt<<1|1]) rmax[rt]=rmax[rt<<1|1]+rmax[rt<<1]; // cout<<l<<" "<<r<<" "<<lmax[rt]<<" "<<rmax[rt]<<endl; } else lmax[rt]=rmax[rt]=1; } int query(int x,int l,int r,int rt) { if(l>x) { if(l-x>rmax[rt-1])return 0; else return lmax[rt]; } else if(r<x) { if(x-r>lmax[rt+1])return 0; else return rmax[rt]; } if(l+lmax[rt]-1>=x)return lmax[rt]; if(r-rmax[rt]+1<=x)return rmax[rt]; return query(x,lson)+query(x,rson); } int num[maxn]; stack<int>st; int main() { int n,m,x; char str[101]; while(~scanf("%d%d",&n,&m)) { creat(root); memset(num,0,sizeof(num)); while(m--) { scanf("%s",str); if(str[0]=='D') { scanf("%d",&x); st.push(x); num[x]++; if(num[x]==1) { update(x,root); } } else if(str[0]=='Q') { scanf("%d",&x); if(num[x]==0)cout<<query(x,root)<<endl; else cout<<"0"<<endl; } else { x=st.top(); st.pop(); if(num[x]!=0) { num[x]=0; update1(x,root); } } } } return 0; }
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。