一个简单的整数问题 - 区间修改和单点查询

给定长度为 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]|≤109

输入样例：

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

#include<bits/stdc++.h>
#define IOS ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
#define endl '\n'

using namespace std;

typedef pair<int, int> PII;
typedef long long ll;

const int N = 100010;

int n, m;
int a[N], tr[N];

inline int lowbit(int x)
{
    return x & -x;
}

inline void add(int x, int c)
{
    for(int i = x; i <= n; i += lowbit(i))
    {
        tr[i] += c;
    }
}

inline int query(int x)
{
    int res = 0;
    for(int i = x; i; i -= lowbit(i))
    {
        res += tr[i];
    }
    return res;
}

int main()
{
    IOS
    cin >> n >> m;
    for(int i = 1; i <= n; i ++)
    {
        cin >> a[i];
        add(i, a[i] - a[i - 1]);
    }
    
    char op[2];
    int l, r, d;
    while(m --)
    {
        cin >> op;
        if(op[0] == 'C')
        {
            cin >> l >> r >> d;
            add(l, d), add(r + 1, -d);
        }
        else
        {
            cin >> l;
            cout << query(l) << endl;
        }
    }
    
    return 0;
}

 区间修改用差分数组来实现，用两次单点修改来造成区间修改的效果。

单点查询用差分数组的前缀和来实现。

学自：Acwing