c++刷题 超能力编程
2040 - 前缀和与差分-差分矩阵
思路
前缀和差分是2个互逆的运算,假设最开始的数组是a[i], 则前缀和数组sum[i]表示从a[1]+…+a[i];而差分数组b[1]+…+b[i]则表示a[i],即a[i]是差分数组b[i]的前缀和;
所以b[i][j]表示 b[1][1]+…+b[i][j]从图上刚好构成一个矩阵,
所以当a[i][j]+c,从b数组构成的矩阵上来看就是b[i][j]这一个小矩阵加上c,画出图之后比较容易可以看出公式为:b[i][j] += c, b[i + 1][j] -= c, b[i][j + 1] -= c, b[i + 1][j + 1] += c;
C++ 代码
#include <iostream>
#include <cstdio>
using namespace std;
const int maxn = 1e3 + 40;
int a[maxn][maxn], b[maxn][maxn];
inline void insert(int x1, int y1, int x2, int y2, int c) {
b[x1][y1] += c;
b[x2 + 1][y1] -= c;
b[x1][y2 + 1] -= c;
b[x2 + 1][y2 + 1] += c;
}
int main(void) {
int n, m, q;
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++)
insert(i, j, i, j, a[i][j]);
for(int i = 1; i <= q; i++) {
int x1, y1, x2, y2, c;
scanf("%d%d%d%d%d", &x1, &y1, &x2, &y2, &c);
insert(x1, y1, x2, y2, c);
}
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++)
if(j == m) printf("%d\n", b[i][j]);
else printf("%d ", b[i][j]);
return 0;
}
作者亲测AC!!!