真的是WA到哭,最后才发现考虑漏掉了一种情况。
题目大意:N 个数,M 次操作(1<=N, M<=100000)。操作分两种情况,一种是将arr[x]更新成y,即arr[x] = y;另一种是询问区间[L, R]中第D位上是P的数有多少个。
解题思路:懂了题意,就应该马上想到线段树,或者树状数组,这也确实正确的思路。不过,仔细想想单点更新区间询问的问题和N和M的范围,就应该想到平方分割也是能够做的。
快上课了,闲话就不多说了,不会平方分割的可以看看《挑战程序设计竞赛》这本书上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) {//之前这里考虑漏了,WA了很多次
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;
}