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\le n,q\le {10}^4,1\le m\le 10,0\le a_i\le {10}^9$。

Input	Output
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;
}