树状数组模板题1,2(做个记录)

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某一个数加上x

2.求出某区间每一个数的和

输入输出格式

输入格式:

第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含3个整数,表示一个操作,具体如下:

操作1: 格式:1 x k 含义:将第x个数加上k

操作2: 格式:2 x y 含义:输出区间[x,y]内每个数的和

输出格式:

输出包含若干行整数,即为所有操作2的结果。

输入输出样例

输入样例#1: 复制

5 5
1 5 4 2 3
1 1 3
2 2 5
1 3 -1
1 4 2
2 1 4

输出样例#1: 复制

14
16

说明

时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:

故输出结果14、16

#include<iostream>
using namespace std;
#define lowbit(i) ((i) & (-i))
const int maxn = 500010;
int c[maxn], A[maxn];

int getSum(int x) {
    int sum = 0;
    for (int i = x; i > 0; i -= lowbit(i)) {
        sum += c[i];
    }
    return sum;
}

void update(int x, int v) {
    for (int i = x; i < maxn; i += lowbit(i)) {
        c[i] += v;
    }
}
int n, m, a, b, d;
int main() {
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> A[i];
        update(i, A[i]);
    }
    for (int i = 1; i <= m; i++) {
        cin >> a >> b >> d;
        if (a == 1) {
            update(b, d);
        } else if (a == 2) {
            long long result = getSum(d) - getSum(b - 1);
            cout << result << endl;
        }
    }
    return 0;
}

题目描述

如题,已知一个数列,你需要进行下面两种操作:

1.将某区间每一个数数加上x

2.求出某一个数的值

输入输出格式

输入格式:

第一行包含两个整数N、M,分别表示该数列数字的个数和操作的总个数。

第二行包含N个用空格分隔的整数,其中第i个数字表示数列第i项的初始值。

接下来M行每行包含2或4个整数,表示一个操作,具体如下:

操作1: 格式:1 x y k 含义:将区间[x,y]内每个数加上k

操作2: 格式:2 x 含义:输出第x个数的值

输出格式:

输出包含若干行整数,即为所有操作2的结果。

输入输出样例

输入样例#1: 复制

5 5
1 5 4 2 3
1 2 4 2
2 3
1 1 5 -1
1 3 5 7
2 4

输出样例#1: 复制

6
10

说明

时空限制:1000ms,128M

数据规模:

对于30%的数据:N<=8,M<=10

对于70%的数据:N<=10000,M<=10000

对于100%的数据:N<=500000,M<=500000

样例说明:

故输出结果为6、10

#include<iostream>
using namespace std;
#define lowbit(i) ((i) & (-i))
const int maxn = 500010;
int c[maxn], A[maxn];
int n, m, a, b, d, e;
int getSum(int x) {
    int sum = 0;
    for(int i = x; i < maxn; i += lowbit(i)) {
        sum += c[i];
    }
    return sum;
}

void update (int x, int v) {
    for (int i = x; i > 0; i -= lowbit(i)) {
        c[i] += v;
    }
}

int main () {
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) {
        cin >> A[i];
        update(i, A[i]);
        update(i - 1, - A[i]);
    }
    for (int i = 1; i <= m; i++) {
        cin >> a;
        if (a == 1) {
            cin >> b >> d >> e;
            update(d, e);
            update(b - 1, -e);
        } else if (a == 2) {
            cin >> b;
            long long result = getSum(b);
            cout << result << endl;
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/LanQiLi/article/details/85225823