Queries Gym - 100741A - 树状数组

给定 \(n\) 和 \(m\)，对于 \(n\) 个数字 \(a_i\)，进行下列三种操作：
（1） + p r： 将 p 位置的元素加上 r， 输出此时 p 位置的值；
（2） - p r ： 将 p 位置的元素减去 r，若 p 位置的值小于r 则不进行减法，输出此时 p 位置的值；
（3） s l r mod：求区间 [l, r] 中值 %m==mod 的所有元素的和，输出该和。

Input

第 \(1\) 行 \(n,m\)；
第 \(2\) 行 \(n\) 个数 \(a_i\)；
第 \(3\) 行 \(q\)；
接下来 \(q\) 行，每行对应一种操作；

Output

输出 \(q\) 行，为每次操作的结果。
数据范围：\(1≤n,q≤10^4, 1≤m≤10, 0≤a_i≤10^9\)。

3 4
1 2 3
3
s 1 3 2
+ 2 1
- 1 2

2
3
1

分析

多个树状数组

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int N=1e4+10,INF=0x3f3f3f3f;
int n,m,q;
LL tree[10][N],a[N];

#define lowbit(x) (x&-(x))
void add(int k,int x,int y){
    while(x<=n) tree[k][x]+=y, x+=lowbit(x);
}
LL sum(int k,int x){
    LL res=0;
    while(x) res+=tree[k][x], x-=lowbit(x);
    return res;
}
int main(){
    // freopen("data.in", "r", stdin);
    cin>>n>>m;
    for(int i=1; i<=n; i++)cin>>a[i], add(a[i]%m, i, a[i]);
    cin>>q; char op; int p,l,r,mod;
    while(q--){
        cin>>op;
        if(op=='+'||op=='-'){
            cin>>p>>r;
            if(op=='-') r*=-1;
            if(a[p]+r >= 0) {
                add((a[p]%m+m)%m, p, -a[p]);
                a[p] += r;
                add((a[p]%m+m)%m, p, a[p]);
            }
            cout<<a[p]<<endl;
        }else{
            cin>>l>>r>>mod;
            cout<<sum(mod, r)-sum(mod,l-1)<<endl;
        }
    }
    return 0;
}