HDU -5057(分块法)

题目链接： Argestes and Sequence

题意

询问你很多区间，问你这一段区间内总共有几个第D位是P的数字。

思路

比赛的时候用三维线段树超内存了，恰逢最近在学习莫队，正好这个题目也是分块的思想，不得不说好嗨哦。看下数据范围，我们开一个sqrt（N）大小的block,每个块也就几百个元素，我们把这个块内所有的信息都收集到block[i][D][P],代表第i块第D位是P 的数字个总个数。这样当我们询问一个区间的时候，如果这两个端点相隔较远，那么他们中间的块就可以直接加起来。

#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int block[350][11][11];
int len = 350;
int a[100001],n,m;
int d1[13]={0,1,10,100,1000,10000,100000,1000000,10000000,100000000,1000000000};
bool elect(int x,int d,int p)
{
    return (x/d1[d])%10==p;
}
int query(int l,int r,int d,int p)
{
    int i;
    int d1 = l/len + 1, d2 = r/len + 1,sum = 0;
    if(d1 == d2)
    {
       for(i = l ; i <= r ; i ++)
       {
           if(elect(a[i],d,p))
            sum ++;
       }
    }
    else{
        for(i = d1+1 ; i <= d2-1 ; i++)
        {
            sum += block[i][d][p];
        }
        for(i = l ; i/len + 1 < d1+1 && i <= n ; i++)
        {
            if(elect(a[i],d,p))
                sum++;
        }
        for(i = r ; i/len +1 > d2 - 1 ; i--)
        {
            if(elect(a[i],d,p))
                sum ++;
        }
    }
    return sum;
}
void update(int x,int y)         //更新x为Y
{
    int d1 = x/len + 1,i,t;
    t = a[x];
    for(i = 1 ;i <= 10 ; i++)
    {
        block[d1][i][t%10]--;
        t /= 10;
    }
    t = y;
    for(i = 1 ; i<= 10 ; i++)
    {
        block[d1][i][t%10]++;
        t /= 10;
    }
    a[x] = y;
}
int main()
{
    int i,l,r,A,B,d,p,t,id,j,T;
    char s[5];
    scanf("%d",&T);
    while(T--)
    {
        memset(block,0,sizeof(block));
        scanf("%d %d",&n,&m);
        for(i = 1;i <= n ; i++)
        {
            scanf("%d",&a[i]);
            id = i / len + 1;
            t = a[i];
            for(j = 1; j <= 10 ; j++)
            {
                block[id][j][t%10]++;
                t /= 10;
            }
        }
        for(i = 1; i <= m ; i++)
        {
            scanf("%s",s);
            if(s[0] == 'Q')
            {
                scanf("%d %d %d %d",&l,&r,&d,&p);
                printf("%d\n",query(l,r,d,p));
            }
            else
            {
                scanf("%d %d",&A,&B);
                update(A,B);
            }
        }
    }
}