基础算法学习之路（三）

前缀和

前缀和可以理解成一个数组中从第一项加到第n项的和
前缀和的作用：可以计算从第l项到第r项的和。
前缀和能把n次计算简化成一次

#include <iostream>
using namespace std;

const int N = 100010;
int a[N],s[N];
int n,m;

int main(){

    scanf("%d%d",&n,&m);
    for(int i = 1; i<= n;i++) scanf("%d",&a[i]);
    //s[0] = 0;
    for(int i = 1;i <= n; i++) s[i] = s[i-1] + a[i]; //前缀和的初始化

    while(m--) {
        int l,r;
        scanf("%d%d",&l,&r);
        printf("%d",s[r] - s[l-1]);//区间和的计算
    }
    return 0;
}

子矩阵的和（二维前缀和）

思路：一图胜千言

就是将区域进行拆分，类似于概率论中的二维随机变量的分布函数那个图哈哈哈。

#include <iostream>

using namespace std;

const int N = 1010;
int m, n, q;
int a[N][N], s[N][N];

int main() {
    scanf("%d%d%d", &n, &m, &q);
    for (int i = 1; i <= n; i++)//从第一个开始存
        for (int j = 1; j <= m; j++)
            scanf("%d", &a[i][j]);

    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];  //求前缀和（根据图形）

    while (q--) {
        int x1, y1, x2, y2;
        scanf("%d%d%d%d", &x1, &y1, &x2, &y2);
        //计算部分和
        printf("%d\n", s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1]);
    }
    return 0;
}

时间复杂度小课堂

O(n)就说明 n个数据要运算n次
O(log n)说明 n个数据要运算log以2为底的n次
所以logn快得多，100万个数据，二分法O(logn)只需要20多次计算！！所以logn快得多，100万个数据，二分法O(logn)只需要20多次计算！！

一维差分

假设有两个数组a，b，使得a[i] = b[1]+b[2]+…+b[i]
则称a是b的前缀和数组，b是a的差分数组。
要在a数组de[l，r]区间内每一个元素都加上c，只需要将b[l]+c,b[r+1]-c即可。

默认a，b数组初始元素为零。不用考虑构造的过程，只用考虑更新的过程

#include <iostream>

using namespace std;

const int N = 100010;
int n, m;
int a[N], b[N];//a表示原数组，b表示差分数组

//b数组当做假想的数组,满足a是b的前缀和，不用考虑构造的过程，只用考虑更新的过程

void insert(int l, int r, int c) {
    b[l] += c;//l之后的每一个元素都加上了c
    b[r + 1] -= c;//r+1之后每一个元素都减去c
    //最终只有[l,r]内的元素加上了c，其余元素不变
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i++) scanf("%d", &a[i]);

    //将a,b数组默认初始全为零，然后在每个区间[i,i]中插入a[i]
    for (int i = 1; i <= n; i++) insert(i, i, a[i]);
    //执行完这一步操作后，b中的元素与a中元素相同

    while (m--) {//进行m次插入操作
        int l, r, c;
        scanf("%d%d%d", &l, &r, &c);
        insert(l, r, c);//在b数组的[l,r]区间插入c
    }

    for (int i = 1; i <= n; i++) b[i] += b[i - 1];  //将b数组变成自己的前缀和数组，操作完之后b是a的前缀和数组+c
    for (int i = 1; i <= n; i++) printf("%d", b[i]);

    return 0;
}

二维差分

b[x1][y1] += c;
b[x2+1][y1] -= c;
b[x1][y2+1] -= c;
b[x2+1][y2+1] += c;

# include <iostream>

using namespace std;

const int N = 1010;
int n, m, q;
int a[N][N], b[N][N];

void insert(int x1, int y1, int x2, int y2, int c) {
    b[x1][y1] += c;
    b[x1][y2 + 1] -= c;
    b[x2 + 1][y1] -= c;
    b[x2 + 1][y2 + 1] += c;
}

int main() {
    scanf("%d%d%d", &n, &m, &q);
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            scanf("%d", &a[i][j]);

        //插入a数组
    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= m; j++)
            insert(i, j, i, j, a[i][j]);

        //在[x1,y1]-[x2,y2]之间加上c,进行q次操作
    while (q--) {
        int x1, y1, x2, y2, c;
        cin >> x1 >> y1 >> x2 >> y2 >> c;
        insert(x1,y1,x2,y2,c);
    }

    //求b数组本身的前缀和
    for(int i = 1;i<=n;i++)
        for(int j = 1;j<= m;j++)
            b[i][j] += b[i-1][j] + b[i][j-1] - b[i-1][j-1];

    for(int i = 1;i<=n;i++) {
        for (int j = 1; j <= m; j++)
            printf("%d ", b[i][j]);
        printf("\n");
    }
}