[树状数组] [前缀和]一个简单的整数问题

一个简单的整数问题

给定长度为N的数列A,然后输入M行操作指令。
第一类指令形如“C l r d”,表示把数列中第l~r个数都加d。
第二类指令形如“Q X”,表示询问数列中第x个数的值。
对于每个询问,输出一个整数表示答案。
输入格式
第一行包含两个整数N和M。
第二行包含N个整数A[i]。
接下来M行表示M条指令,每条指令的格式如题目描述所示。
输出格式
对于每个询问,输出一个整数表示答案。
每个答案占一行。
数据范围
1≤N,M≤105,
|d|≤10000,
|A[i]|≤1000000000
输入样例:
10 5
1 2 3 4 5 6 7 8 9 10
Q 4
Q 1
Q 2
C 1 6 3
Q 2
输出样例:
4
1
2
5


这题目的关键是如何处理区间加d。
我们用

add(l, d);
add(r + 1, -d);

把l后面的都加d,把r+1后面的都减去d。这样就可以实现[l, r]的区间+d。


#include
using namespace std;
const int maxn = 1e5 + 5;
long long  tree[maxn];
int w[maxn];

int n, m;
int lowbit(int x) {
    return x & (-x);
}

void add(int num, int x) {
    for (int i = num; i <= n; i += lowbit(i)) {
        tree[i] += x;
    }
}

long long  get_sum(int num) {
    long long  ans = w[num];
    for (int i = num; i; i -= lowbit(i)) {
        ans += tree[i];
    }
    return ans;
}

int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> w[i];
    }
    char c;
    for (int i = 1; i <= m; i++) {
        cin >> c;
        if (c == 'C') {
            int l, r, d;
            cin >> l >> r >> d;
            add(l, d);
            add(r + 1, -d);
        }
        if (c == 'Q') {
            int x;
            cin >> x;
            cout << get_sum(x) << "\n";
        }
    }
    return 0;
}

你可能感兴趣的:(蓝桥杯)