HDU - 1540 Tunnel Warfare 超详细图文题解(线段树 单点查询+连续区间)

题目链接

题目大意

在抗日战争期间，地道战在华北平原得到广泛的实施，一般而言，村庄通过一些隧道在一条线上连接，除了两端剩下的每个村庄都有两个相连。侵略者会频繁的对这些村庄进行扫荡，并且摧他们的地道，当然八路军会把这一些已经被摧毁的村庄修复的，会优先修复最近被破坏的村庄。输入一个村庄编号（1-n），求与该村庄连接的村庄个数（包括自己）。

思路

1. 用一个栈模拟村庄的破坏和修复
2. 利用线段树 求出每个线段左端点长度lsum（自左端点向右的连续长度），右端点长度rsum（自右端点向左连续长度） 以及区间内最长连续长度（以为可能存在一个连续长度既不包含左端点也不包含右端点）
3. 破坏和修复用同一个函数，1表示存在，0表示被破坏
4. 查询具体看代码

AC代码

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

const int maxn = 50005;

struct node {
    int L, R, lsum, rsum, sum;
    int Mid() {return (L+R)/2;}
    int Len() {return (R-L+1);}
}a[maxn*4];
void pushup(int p) {
    a[p].lsum = a[p << 1].lsum;
    a[p].rsum = a[p << 1|1].rsum;
    if (a[p << 1].lsum == a[p << 1].Len())
        a[p].lsum = a[p << 1].lsum+a[p << 1|1].lsum;
    if (a[p << 1|1].rsum == a[p << 1|1].Len())
        a[p].rsum = a[p << 1|1].rsum+a[p << 1].rsum;
    a[p].sum = max(a[p].lsum, max(a[p].rsum, a[p << 1].rsum+a[p << 1|1].lsum));
}

void Build(int p, int l, int r) {
    a[p].L = l;
    a[p].R = r;
    a[p].lsum = a[p].rsum = a[p].sum = a[p].Len();
    if (l == r) return;
    Build(p << 1, l, a[p].Mid());
    Build(p << 1|1, a[p].Mid()+1, r);
}

void Insert(int p, int k, int x) {
    if (a[p].L == a[p].R) {
        a[p].lsum = a[p].rsum = a[p].sum = x;
        return;
    }
    if (k <= a[p].Mid())
    Insert(p << 1, k, x);
    else
    Insert(p << 1|1, k, x);
    pushup(p);
}

int Query(int p, int k) {
    if (a[p].sum == 0) return 0;
    if (k < a[p].L+a[p].lsum) return a[p].lsum;
    if (k > a[p].R-a[p].rsum) return a[p].rsum;
    if (k > a[p << 1].R-a[p << 1].rsum&&k < a[p << 1|1].L+a[p << 1|1].lsum)
    return a[p << 1].rsum+a[p << 1|1].lsum;
    if (k <= a[p].Mid())
        return Query(p << 1, k);
    else
        return Query(p << 1|1, k);
}
int main() {
    int n, m;
    while (~scanf("%d%d", &n, &m)) {
        int x;
        char s[10];
        stack <int> sta;
        Build(1, 1, n);
        while (m--) {
            scanf("%s", s);
            if (s[0] == 'D') {
                scanf("%d", &x);
                Insert(1, x, 0);
                sta.push(x);
            } else if (s[0] == 'R'&&sta.size()) {
                Insert(1, sta.top(), 1);
                sta.pop();
            } else if (s[0] == 'Q') {
                scanf("%d", &x);
                printf("%d\n", Query(1, x));
            }
        }
    }
    return 0;
}

图片模拟过程

我一开始也不会这道题，看了别人的代码也懵，自己画图模拟了一遍就理解了，同样迷惑的，可以自行模拟一遍，大佬自动忽略~(dl应该也不会点进来)

下图模拟了建树 破坏 修复的过程，这几个搞明白了查询结合代码也很清晰了