HDU 4267 A Simple Problem with Integers 数列离散化 + 树状数组

题意： 对一个长度为n（n <= 50000）的数列A进行m（m <= 50000)次操作（2种）：
         1：给定a，b，k，c四个整数,在[a , b] 区间满足a<=i<=b && （i-a）% k == 0的数加c（1<=k<=10）；
         2：给定整数a，求当前A[a]；
题解：想办法把间隔的数字搞到一起，因为1<=k<=10，所以一共最多只能形成55个个序列，即：
          1,2,3,4,5…（k=1）
          1,3,5,7,9…（k=2）
          2,4,6,8,10…（k=2）
          1,4,7,10,13…（k=3）
          2,5,9,12,15…（k=3）
          3,6,10,13,16…（k=3）
          …
          1,11,21,31,41…（k=10）
          …
          10,20,30,40,50...（k=10）
          想象中将这些序列按照k不同离散化，然后用树状数组维护区间修改，点查询的时候要枚举k即可。

Sure原创，转载请注明出处。

#include <iostream>
#include <cstdio>
#include <memory.h>
using namespace std;
const int maxn = 50002;
const int maxm = 12;
int list[maxn],c[maxn][maxm][maxm];
int m,n;

inline void in(int &a)
{
    bool flag = true;
    char ch;
    while(ch = getchar(), ch < '0' || ch > '9')
    {
        if(ch == '-') flag = false;
    }
    a = ch - '0';
    while(ch = getchar(), ch >= '0' && ch <= '9')
    {
        a = a * 10 + ch - '0';
    }
    if(flag == false) a = -a;
    return;
}

inline int lowbit(int x)
{
    return x & (-x);
}

void update(int i,int j,int k,int val,int lim)
{
    while(i <= lim)
    {
        c[i][j][k] += val;
        i += lowbit(i);
    }
    return;
}

int sum(int i,int j,int k)
{
    int res = 0;
    while(i > 0)
    {
        res += c[i][j][k];
        i -= lowbit(i);
    }
    return res;
}

void read()
{
    memset(c,0,sizeof(c));
    for(int i=1;i<=n;i++)
    {
        in(list[i]);
    }
    return;
}

void solve()
{
    int tmp,a,b,k,val;
    in(m);
    while(m--)
    {
        in(tmp);
        if(tmp == 1)
        {
            in(a),in(b),in(k),in(val);
            int wei = a % k + 1;
            b += k - ((b - a) % k);
            a = (a - 1) / k + 1;
            b = (b - 1) / k + 1;
            update(a , k , wei , val , n / k + (n % k != 0));
            update(b , k , wei , -val , n / k + (n % k != 0));
        }
        else
        {
            in(a);
            int res = list[a];
            for(int i=1;i<=10;i++)
            {
                int wei = a % i + 1;
                int st = (a - 1) / i + 1;
                res += sum(st , i , wei);
            }
            printf("%d\n",res);
        }
    }
    return;
}

int main()
{
    while(~scanf("%d",&n))
    {
        read();
        solve();
    }
    return 0;
}