HDU 5057 Argestes and Sequence(平方分割)

真的是ＷＡ到哭，最后才发现考虑漏掉了一种情况。

题目大意：N 个数，M 次操作（1<=N, M<=100000）。操作分两种情况，一种是将arr[x]更新成ｙ，即arr[x] = y；另一种是询问区间[L, R]中第Ｄ位上是P的数有多少个。

解题思路：懂了题意，就应该马上想到线段树，或者树状数组，这也确实正确的思路。不过，仔细想想单点更新区间询问的问题和Ｎ和Ｍ的范围，就应该想到平方分割也是能够做的。

快上课了，闲话就不多说了，不会平方分割的可以看看《挑战程序设计竞赛》这本书上P183，讲的很明白。
Show me the code!

#include <iostream>
#include <cmath>
using namespace std;
const int maxn = 1e5;
int arr[maxn], block[400][10][10];
void clear() {
    fill(arr, arr + maxn, 0);
    for (int a = 0; a < 400; ++a) {
        for (int b = 0; b < 10; ++b) {
            fill(block[a][b], block[a][b] + 10, 0);
        }
    }
}
int main() {
    ios::sync_with_stdio(false);
    int T;
    cin >> T;
    while (T--) {
        clear();
        int N, M;
        cin >> N >> M;
        int size = sqrt(N);
        for (int a = 0; a < N; ++a) {
            cin >> arr[a];
            for (int b = 0, t = arr[a]; b < 10; ++b, t /= 10) {
                block[a / size][b][t % 10] += 1;
            }
        }
        while (M--) {
            char op;
            cin >> op;
            if (op == 'S') {
                int x, y;
                cin >> x >> y;
                x -= 1;
                for (int a = 0, t = arr[x]; a < 10; ++a, t /= 10) {
                    block[x / size][a][t % 10] -= 1;
                }
                for (int a = 0, t = y; a < 10; ++a, t /= 10) {
                    block[x / size][a][t % 10] += 1;
                }
                arr[x] = y;
            }
            else {
                int l, r, d, p;
                cin >> l >> r >> d >> p;
                l -= 1, r -= 1, d -= 1;
                int res = 0;
                if (l / size == r / size) {//之前这里考虑漏了，ＷＡ了很多次
                    for (int a = l; a <= r; ++a) {
                        for (int b = 0, t = arr[a]; b <= d; ++b, t /= 10) {
                            if (b == d && t % 10 == p) res += 1;
                        }
                    }
                    cout << res << endl;
                    continue;
                }
                for (int a = l; a < (l / size + 1) * size; ++a) {
                    for (int b = 0, t = arr[a]; b <= d; ++b, t /= 10) {
                        if (b == d && t % 10 == p) res += 1;
                    }
                }
                for (int a = l / size + 1; a < r / size; ++a) {
                    res += block[a][d][p];
                }
                for (int a = r / size * size; a <= r; ++a) {
                    for (int b = 0, t = arr[a]; b <= d; ++b, t /= 10) {
                        if (b == d && t % 10 == p) res += 1;
                    }
                }
                cout << res << endl;
            }
        }
    }
    return 0;
}