hdu 7267

DickensTone

于 2018-02-11 20:08:34 发布

阅读量161

点赞数

CC 4.0 BY-SA版权

文章标签：树状数组

本文链接：https://blog.youkuaiyun.com/DickensTone/article/details/79312993

本文介绍了一种基于树状数组的数据结构实现，用于处理序列的区间修改与查询问题。特别是针对那些满足特定间隔条件k的操作，提供了高效算法。通过实例演示了如何在一系列查询和更新中维护序列状态。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：对一段序列进行查询和修改。其中有限制条件k。

修改i ([a, b])序列时，要满足 (i - a) % k == 0。换句话说，就是要每k个数后修改。那么我们只需要保证 % k的值相同。

#include<cstdio>
#include<cstring>
const int maxn = 50000 +5;
int tree[11][11][maxn], n;
int A[maxn];
void Add(int k, int mod, int a, int val)
{
    while(a <= n)
    {
        tree[k][mod][a] += val;
        a += a & -a;
    }
}
int Query(int k, int mod, int a)
{
    int ret = 0;
    while(a > 0)
    {
        ret += tree[k][mod][a];
        a -= a & -a;
    }
    return ret;
}
int main()
{
    while(scanf("%d", &n) == 1)
    {
        for(int i = 1; i <= n; i++)
            scanf("%d", &A[i]);
        memset(tree, 0, sizeof(tree));
        int Q;
        scanf("%d", &Q);
        while(Q--)
        {
            int cmd;
            scanf("%d", &cmd);
            if(cmd == 1)
            {
                int a, b, k, c;
                scanf("%d%d%d%d", &a, &b, &k, &c);
                // a -> b 从a开始，每隔k个数，修改。就是 %k的值相同、
                Add(k, a % k, a, c);
                Add(k, a % k, b + 1, -c);
            }
            else if(cmd == 2)
            {
                int a;
                scanf("%d", &a);
                int ans = A[a];
                for(int k = 1; k <= 10; k++)
                {
                    ans += Query(k, a % k, a);
                }
                printf("%d\n", ans);
            }
        }
    }

    return 0;
}