华华开始学信息学

树状数组 设立一个阈值

链接：https://ac.nowcoder.com/acm/contest/392/F
来源：牛客网
 

题目描述

因为上次在月月面前丢人了，所以华华决定开始学信息学。十分钟后，他就开始学树状数组了。这是一道树状数组的入门题：

给定一个长度为N的序列A，所有元素初值为0。接下来有M次操作或询问：
操作：输入格式：1 D K，将ADAD加上K。
询问：输入格式：2 L R，询问区间和，即∑Ri=LAi∑i=LRAi。

华华很快就学会了树状数组并通过了这道题。月月也很喜欢树状数组，于是给华华出了一道进阶题：
给定一个长度为N的序列A，所有元素初值为0。接下来有M次操作或询问：
操作：输入格式：1 D K，对于所有满足1≤i≤N1≤i≤N且i≡0(modD)i≡0(modD)的i，将AiAi加上K。
询问：输入格式：2 L R，询问区间和，即∑Ri=LAi∑i=LRAi。
华华是个newbie，怎么可能会这样的题？不过你应该会吧。

输入描述:

第一行两个正整数N、M表示序列的长度和操作询问的总次数。
接下来M行每行三个正整数，表示一个操作或询问。

输出描述:

对于每个询问，输出一个非负整数表示答案。

示例1

输入

10 6
1 1 1
2 4 6
1 3 2
2 5 7
1 6 10
2 1 10

输出

3
5
26

备注:

1≤N,M≤1051≤N,M≤105，1≤D≤N1≤D≤N，1≤L≤R≤N1≤L≤R≤N，1≤K≤108

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 5;
const int s = 500;
long long a[maxn], extra[s];
int n, m;
int lowbit(int x){return x&(-x);}
void update(int k,int x)
{
    for(int i = k; i <= n; i += lowbit(i)){
        a[i] += x;
    }
}
long long getnum(int x)
{
    long long ans = 0;
    for(int i = x; i; i -= lowbit(i)){
        ans += a[i];
    }
    return ans;
}
int main()
{
    scanf("%d%d",&n,&m);
    int op, x, y;
    for(int i = 1; i <= m; i++){
        scanf("%d%d%d",&op,&x,&y);
        if(op == 1){
            if(x < s) extra[x] += y;
            else{
                for(int j = x; j <= n; j += x){
                    update(j,y);
                }
            }
        }
        else{
            long long ans = getnum(y)-getnum(x-1);
            for(int j = 1; j < s; j++){
                ans += extra[j]*(y/j-(x-1)/j);
            }
            printf("%lld\n",ans);
        }
    }
    return 0;
}