ccf201709-5除法(树状数组)

最新推荐文章于 2024-04-29 21:49:08 发布

原创最新推荐文章于 2024-04-29 21:49:08 发布 · 480 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#ccf / shu zhuang shu zu

本文介绍了一种使用二进制索引树（BIT）解决区间修改与查询问题的方法。具体实现包括更新指定区间内特定数值的倍数并求解指定区间的和。通过对每个操作的高效处理，该算法能够应对大规模数据集的挑战。

问题描述

　　小葱喜欢除法，所以他给了你N个数a1, a2, ⋯, aN，并且希望你执行M次操作，每次操作可能有以下两种：
　　给你三个数l, r, v，你需要将al, al+1, ⋯, ar之间所有v的倍数除以v。
　　给你两个数l, r，你需要回答al + al+1 + ⋯ + ar的值是多少。

输入格式

　　第一行两个整数N, M，代表数的个数和操作的次数。
　　接下来一行N个整数，代表N个数一开始的值。
　　接下来M行，每行代表依次操作。每行开始有一个整数opt。如果opt=1，那么接下来有三个数l, r, v，代表这次操作需要将第l个数到第r个数中v的倍数除以v；如果opt = 2，那么接下来有两个数l, r，代表你需要回答第l个数到第r个数的和。

输出格式

　　对于每一次的第二种操作，输出一行代表这次操作所询问的值。

样例输入

5 3
1 2 3 4 5
2 1 5
1 1 3 2
2 1 5

样例输出

15
14

评测用例规模与约定

　　对于30%的评测用例，1 ≤ N, M ≤ 1000；
　　对于另外20%的评测用例，第一种操作中一定有l = r；
　　对于另外20%的评测用例，第一种操作中一定有l = 1 , r = N；

　　对于100%的评测用例，1 ≤ N, M ≤ 105，0 ≤ a1, a2, ⋯, aN ≤ 106, 1 ≤ v ≤ 106, 1 ≤ l ≤ r ≤ N。

利用二进制的lowbit进行构建

#include<iostream>
#include<cstdio>

using namespace std;

typedef long long ll;

ll tree[101024];
int n,m,a[101024];
//核心
void update(int i,int val)
{
    while(i<=n){
        tree[i]+=val;
        i+=-i&i;
    }
}
//核心
ll getsum(int i)
{
    ll sum=0;
    while(i>0){
        sum+=tree[i];
        i-=-i&i;
    }
    return sum;
}

int main()
{
    int opt,l,r,w;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        update(i,a[i]);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&opt,&l,&r);
        if(opt==2){
            printf("%lld\n",getsum(r)-getsum(l-1));
        }else{
            scanf("%d",&w);
            if(w==1){
                continue;
            }
            while(l<=r){
                if(a[l]>=w&&a[l]%w==0){
                    update(l,-(a[l]-a[l]/w));
                    a[l]/=w;
                }
                ++l;
            }
        }
    }
    return 0;
}