题意
有一组数,支持两种操作,修改和查询。每次查询可查询L到R这个区间内,第D位为P的元素有多少个。每次修改可修改一个点的值。
题解
纯分块,将元素分成256个元素为一块。使用num[i][j][k]代表第i块,第j位为k的元素有多少个。dig[i][j]代表第i个元素的第j位为多少。每次修改直接修改就可以了。每次查询对于l所在块和r所在块逐个遍历,对于中间块直接读取值就可以了。
代码
#include <iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<string>
#include<set>
#include<map>
#include<bitset>
#include<stack>
#include<string>
#define UP(i,l,h) for(int i=l;i<h;i++)
#define DOWN(i,h,l) for(int i=h-1;i>=l;i--)
#define W(a) while(a)
#define MEM(a,b) memset(a,b,sizeof(a))
#define LL long long
#define INF 0x3f3f3f3f
#define MAXN 1010
#define MOD 1000000007
#define EPS 1e-3
using namespace std;
int num[400][10][10];
int dig[100010][10];
int main() {
int t;
scanf("%d",&t);
W(t--) {
MEM(num,0);
MEM(dig,0);
int n,m;
scanf("%d%d",&n,&m);
UP(i,1,n+1) {
int x;
scanf("%d",&x);
UP(j,0,10) {
num[i>>8][j][dig[i][j]=x%10]++;
x/=10;
}
}
W(m--) {
char op[5];
scanf("%s",op);
if(strcmp(op,"Q")==0) {
int l,r,d,p,ans=0;
scanf("%d%d%d%d",&l,&r,&d,&p);
d--;
int lb=(l>>8)+1,rb=r>>8;
// cout<<lb<<" "<<rb<<endl;
if(lb<=rb) {
int lx=lb<<8,rx=rb<<8;
for(; l<lx; l++) ans+=dig[l][d]==p;
for(; lb<rb; lb++) ans+=num[lb][d][p];
for(; rx<=r; rx++) ans+=dig[rx][d]==p;
} else {
for(; l<=r; l++) ans+=dig[l][d]==p;
}
printf("%d\n",ans);
} else {
int x,y;
scanf("%d%d",&x,&y);
UP(j,0,10) {
num[x>>8][j][dig[x][j]]--;
num[x>>8][j][dig[x][j]=y%10]++;
y/=10;
}
}
}
}
}