HDU 5057 Argestes and Sequence（树状数组）

题目描述：

给你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;
}