HDU 1540 Tunnel Warfare(线段树，记得这题是多组输入)

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

解题思路：

线段树维护区间左端点向右可以延伸多少，右端点可以向左延伸多少

每次询问x，求出 [1,x-1] 和 [x,n] ，答案就是第一个区间的右向左+第二个区间的左向右

注意这题是多组输入。

重难点在query函数，这个能看懂其他地方肯定没问题了

代码：

#include<cstdio>
#include<algorithm>
#include<cstring>
#define lson rt<<1,l,m
#define rson rt<<1|1,m+1,r
#define mid int m = l+r>>1
#define ll long long
#define tl tree[rt<<1]
#define tr tree[rt<<1|1]
#define debug(x) printf("Line:%s----\n",#x)

using namespace std;

const int N = 5e4+5;

struct T
{
    int l,r,len;
    T(){l=r=len=0;}
}tree[N<<2];
int pre[N],idx;///储存摧毁的点

void push_up(int rt)
{
    tree[rt].l = (tl.l==tl.len)? tl.len + tr.l : tl.l;///
    tree[rt].r = (tr.r==tr.len)? tr.len + tl.r : tr.r;
    tree[rt].len  = tl.len + tr.len;
}

void update(int p,int x,int rt,int l,int r)
{
    if (l==r){
        tree[rt].l = tree[rt].r = x;
        return ;
    }
    mid;
    if (p<=m) update(p,x,lson);
    else update(p,x,rson);
    push_up(rt);
}

T query(int L,int R,int rt,int l,int r)
{
    if (L==l && r==R){
        return tree[rt];
    }
    mid;
    T a,b,ans;
    if (L > m) return query(L,R,rson);     ///就这边要注意吧
    else if (R<= m) return query(L,R,lson);
    a = query(L,m,lson);
    b = query(m+1,R,rson);
    ans.l = (a.l==a.len)? a.len+b.l : a.l;
    ans.r = (b.r==b.len)? b.len+a.r : b.r;
    ans.len = a.len + b.len;
    return ans;
}

void build(int rt,int l,int r)
{
    if (l==r){
        tree[rt].l = tree[rt].r = 1;
        tree[rt].len = 1;
        return ;
    }
    mid;
    build(lson);
    build(rson);
    push_up(rt);
}
/*//可以看一下线段树维护的数据
void pt(int rt,int l,int r)
{
    if (l==r){
        printf("t[%d]->[%d,%d],t.l=%d,t.r=%d\n",rt,l,r,tree[rt].l,tree[rt].r);
        return ;
    }
    mid;
    pt(lson);
    pt(rson);
    printf("t[%d]->[%d,%d],t.l=%d,t.r=%d\n",rt,l,r,tree[rt].l,tree[rt].r);
}
*/
int main()
{
    int n,q;
    while(~scanf("%d %d",&n,&q)){
        build(1,1,n);
        //pt(1,1,n);
        idx=0;
        char op[2];
        int x;
        while (q--){
            scanf("%s",op);

            if (op[0]=='D'){
                scanf("%d",&x);
                update(x,0,1,1,n);
                pre[++idx] = x;
                //pt(1,1,n);
            }
            else if (op[0]=='R'){
                if (idx==0) continue;
                update(pre[idx--],1,1,1,n);
                //pt(1,1,n);
            }
            else if (op[0]=='Q'){
                scanf("%d",&x);
                if (x==1) printf("%d\n",tree[1].l);
                else if (x==n) printf("%d\n",tree[1].r);
                else {
                    T a,b;
                    a = query(1,x-1,1,1,n); 
                    b = query(x,n,1,1,n);   
                    if (b.l==0) printf("0\n");
                    else printf("%d\n",a.r+b.l);
                }
            }
        }
    }
    return 0;
}

debug历程：

1.一开始结构体中没有len表示当前树的区间长度，用r-l+1来求左子树右子树区间长度，由于麻烦宏定义了#define len r-l+1

然后左子树区间长度 = len-len/2

       右子树区间长度 = len/2

搞了半天发现应该#define len （r-l+1）

后来想着这样算太麻烦，直接记录区间长度得了。

2.query函数中按照区间求和的方法

if (L<=l && r<=R){

        return tree[rt];

}

mid;

T a,b,ans;

if (L<=m) a=query(L,R,lson);

if (R>m) b=query(L,R,rson);

再合并给ans

这样求出来的a，b是什么东西？反正不是想要的。求出来的a，b应该不是精准的区间，只是在[L,R]区间内就会返回

于是改成了如上代码中这样

3.WA了之后自己试了下数据，发现了Q 1直接程序崩溃，然后发现主函数中query(1,0,1,1,n)然后想到1，n特殊处理

4.错误在哪呢，错误在哪呢？翻别人题解？这不差不多吗，哪错了？？？

几个小时后：“cao，多组输入！”