题目描述:
给你N个数,有M个查询,每次查询输入一个字符“Q”、“S”,为“Q”时输入四个数,“L”、“R”、“D”、“P”,代表查询这N个数从“L"到”R“之间第D位(从个位数起)为”P“的个数。为”S“时,输入两个数,”X“、”Y“,表示将这N个数中的第”X“个数,改为”Y“。
做题想法:
刚开始的时候开了一个三维 的树状数组,结果MLE了,后来就想到每个数的各个为互不影响,就可以从第一位到第十位(2^31不超过10位)分别处理,每一位存储在树状数组中,压缩为二维。
首先将这N个数存储在数组中,再将M次查询记录在一个结构体中。在处理每一位中,遍历M次操作,判断是否有在此位进行的操作。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int c[100005][15];
int n;
int a[100005];
int vis[100005];
struct node
{
int flag,L,R,D,P,x,y,ans;
}p[100005];
int lowbit(int x)
{
return x&(-x);
}
void update(int i,int num,int x)
{
while(i<=n)
{
c[i][num]+=x;
i+=lowbit(i);
}
}
int sum(int num,int i)
{
int ans=0;
while(i>0)
{
ans+=c[i][num];
i-=lowbit(i);
}
return ans;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int M;
scanf("%d%d",&n,&M);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
}
char op[10];
for(int i=1;i<=M;i++)
{
scanf("%s",op);
if(op[0]=='S')
{
p[i].flag=0;
scanf("%d%d",&p[i].x,&p[i].y);
}
else
{
p[i].flag=1;
scanf("%d%d%d%d",&p[i].L,&p[i].R,&p[i].D,&p[i].P);
}
}
int temp=1;
for(int i=1;i<=10;i++)
{
memset(c,0,sizeof(c));///存储每一位时初始化树状数组
for(int j=1;j<=n;j++)
{
update(j,a[j]/temp%10,1);///将每个数的第i位存储在树状数组中
vis[j]=a[j]/temp%10;
}
for(int j=1;j<=M;j++) ///在每一位中遍历查询,是否有在此位的进行操作
{
if(p[j].flag==1&&p[j].D==i)
p[j].ans=sum(p[j].P,p[j].R)-sum(p[j].P,p[j].L-1);
else if(p[j].flag==0)
{
update(p[j].x,vis[p[j].x],-1);///要注意改变某一个数时,会对其他结果有影响
update(p[j].x,p[j].y/temp%10,1);///所以要记录其改变的值,而不能直接改变这个数的值
vis[p[j].x]=p[j].y/temp%10;
}
}
temp*=10;
}
for(int i=1;i<=M;i++)
if(p[i].flag==1)
printf("%d\n",p[i].ans);
}
return 0;
}