I - Tunnel Warfare
阿新 • • 發佈：2018-02-25
div gpo gin rip hang 格式 nta 自己 scrip
I - Tunnel Warfare
HDU - 1540 思路：原來以為自己已經完全理解了線段樹，現在發現其實還差一些火候，做題的時候太拘泥於格式，思路不是很能放開。#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define MAXN 50010 using namespace std; int n,m,top; int stack[MAXN],vis[MAXN]; struct nond{int l,r,len,ren,mid; }tree[MAXN*4]; void build(int now,int l,int r){ tree[now].l=l;tree[now].r=r; tree[now].len=tree[now].r-tree[now].l+1; tree[now].ren=tree[now].mid=tree[now].len; if(tree[now].l==tree[now].r) return ; int mid=(tree[now].l+tree[now].r)/2; build(now*2,l,mid); build(now*2+1,mid+1,r); } void change(int now,int x,int pos){ if(tree[now].l==tree[now].r){ tree[now].len=tree[now].ren=tree[now].mid=pos; return ; } int mid=(tree[now].l+tree[now].r)/2; if(x<=mid) change(now*2,x,pos); else if(x>mid) change(now*2+1,x,pos); tree[now].len=tree[now*2].len; tree[now].ren=tree[now*2+1].ren; tree[now].mid=max(tree[now*2].mid,tree[now*2+1].mid); tree[now].mid=max(tree[now].mid,tree[now*2].ren+tree[now*2+1].len); if(tree[now*2].len==tree[now*2].r-tree[now*2].l+1) tree[now].len=tree[now*2].len+tree[now*2+1].len; if(tree[now*2+1].ren==tree[now*2+1].r-tree[now*2+1].l+1) tree[now].ren=tree[now*2+1].ren+tree[now*2].ren; } int query(int now,int k){ if(tree[now].l==tree[now].r||tree[now].mid==0||tree[now].mid==tree[now].r-tree[now].l+1) return tree[now].mid; int mid=(tree[now].l+tree[now].r)/2; if(k<=mid){ if(k>=tree[now*2].r-tree[now*2].ren+1) return query(now*2,k)+query(now*2+1,mid+1); else return query(now*2,k); } else{ if(k<=tree[now*2+1].l+tree[now*2+1].len-1) return query(now*2+1,k)+query(now*2,mid); else return query(now*2+1,k); } } int main(){ while(scanf("%d%d",&n,&m)!=EOF){ build(1,1,n);top=0; for(int i=1;i<=m;i++){ char c;int x; scanf("\n%c",&c); if(c==‘D‘){ scanf("%d",&x); stack[++top]=x;change(1,x,0); } else if(c==‘Q‘){ scanf("%d",&x); printf("%d\n",query(1,x)); } else change(1,stack[top--],1); } } } /* 7 9 D 3 D 6 D 5 Q 4 Q 5 R Q 4 R Q 4 */
I - Tunnel Warfare