[HDU 4046]Panda[线段树]

题目链接： [HDU 4046]Panda[线段树]

题意分析：

在一个长度为n的只包含b和w的字符串中，给出m个操作，0操作改某位为b或者w中的一个字符，1操作查询区间[L,R]内有多少个wbw

解题思路：

sum[i]代表前到第i个字符为止所拥有的wbw的个数，a[]用来存储线段树结点值，也就是sum[]值。整道题就是线段树区间更新和单点查询。

个人感受：

想到用sum来存储，然后推算更新单点会影响后方所有的点，不知道怎么处理（尼玛，其实就是线段树区间更新啊喂）。然后和队友讨论，知道是线段树就开搞了= =。结果也是蛮坑的。首先区间查询的话，实际上是++L，++R后，查询[L + 1，R]这个区间段的值，因为L + 1, L, L - 1有可能构成wbw但是这个却不属于我们的查询区间[L,R]，所以要从L + 1开始。其次，L可以等于R，导致L + 1可能大于R，没注意啊。。。。。唯一值得欣慰的是线段树没写错。。。。。23333

具体代码如下：

#include<algorithm>
#include<cctype>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iomanip>
#include<iostream>
#include<map>
#include<queue>
#include<set>
#include<sstream>
#include<stack>
#include<string>
#define lowbit(x) (x & (-x))
#define root 1, n, 1
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1
#define ll long long
#define pr(x) cout << #x << " = " << (x) << '\n';
using namespace std;

const int INF = 0x7f7f7f7f;
const int MAXN = 5e4 + 111;

int sum[MAXN], a[MAXN << 2], add[MAXN << 2];
char s[MAXN];

void build(int l, int r, int rt) {
    add[rt] = 0;
    if (l == r) {
        a[rt] = sum[l];
        return;
    }
    int m = l + r >> 1;
    build(lson);
    build(rson);
    return;
}

void push_down(int rt) {
    if (add[rt] != 0) {
        add[rt << 1] += add[rt];
        add[rt << 1 | 1] += add[rt];
        add[rt] = 0;
    }
}

void update(int L, int R, int x, int l, int r, int rt) {
    if (L <= l && r <= R) {
        add[rt] += x;
        return;
    }
    push_down(rt);
    int m = l + r >> 1;
    if (L <= m) update(L, R, x, lson);
    if (m < R) update(L, R, x, rson);
    return;
}

int query(int x, int l, int r, int rt) {
    if (l == r) {
        return a[rt] + add[rt];
    }
    push_down(rt);
    int m = l + r >> 1;
    if (m < x) return query(x, rson);
    else return query(x, lson);
}

int main()
{
    int m, t, n;
    for (int kase = scanf("%d", &t); kase <= t; ++kase) {
        printf("Case %d:\n", kase);
        scanf("%d%d%s", &n, &m, s + 1);
        sum[0] = sum[1] = sum[2] = 0;
        for (int i = 3; i <= n; ++i) {
            sum[i] = sum[i - 1];
            if (s[i] == 'w' && s[i - 1] == 'b' && s[i - 2] == 'w')
                ++sum[i];
        }
        build(root);
        int op, l, r;
        char x[2];
        while (m --) {
            scanf("%d%d", &op, &l);
            if (op == 0) {
                scanf("%d", &r);
                ++l, ++r;
                if (r - l + 1 < 3) printf("0\n");
                else
                    printf("%d\n", query(r, root) - query(l + 1, root));
            }
            else {
                scanf("%s", x);
                ++l;
                if (s[l] != x[0]) {
                    if (l >= 3 && s[l] == 'w' && s[l - 1] == 'b' && s[l - 2] == 'w')
                        update(l, n, -1, root);

                    if (l >= 2 && l + 1 <= n && s[l] == 'b' && s[l - 1] == 'w' && s[l + 1] == 'w')
                        update(l + 1, n, -1, root);

                    if (l + 2 <= n && s[l] == 'w' && s[l + 1] == 'b' && s[l + 2] == 'w')
                        update(l + 2, n, -1, root);

                    s[l] = x[0];

                    if (l >= 3 && s[l] == 'w' && s[l - 1] == 'b' && s[l - 2] == 'w')
                        update(l, n, 1, root);

                    if (l >= 2 && l + 1 <= n && s[l] == 'b' && s[l - 1] == 'w' && s[l + 1] == 'w')
                        update(l + 1, n, 1, root);

                    if (l + 2 <= n && s[l] == 'w' && s[l + 1] == 'b' && s[l + 2] == 'w')
                        update(l + 2, n, 1, root);
                }
            }
        }
    }
    return 0;
}