[CODEVS 3147] 矩阵乘法 2

描述

给出两个n*n的矩阵，m次询问它们的积中给定子矩阵的数值和。

分析

直接 n3 的矩阵乘法肯定超时, 要采用前缀和优化

row[s1] … row[t1] col[s2] … col[t2]

(s1, s2) – (t1, t2)

row[x] * col[y] 表示用第 x 行的所有元素去乘第 y 行的所有元素.

==>

= row[s1] * col[s2] + row[s1] * col[s2+1] + … + row[s1] * col[t2] + row[s1+1] * col[s2] + … + row[s1+1] * col[t2] + … + row[t1] * col[t2] // 分配律, 其实并没有看上去那么显然

= row[s1] * (col[s2] + col[s2+1] + … + col[t2]) + row[s1+1] * (col[s2] + col[s2+1] + … + col[t2]) + … + row[t1] * (col[s2] + col[s2+1] + … + col[t2])

= (row[s1] + row[s1+1] + … + row[t1]) * (col[s2] + col[s2+1] + … + col[t2])

==> 用前缀和处理 注意: row 其实相当于一个 1 行 n 列的矩阵, 而 col 相当于一个 n 行 1 列的矩阵. 上式中 row[s1] + … + row[t1] 表示把 t1-s1+1 个这样的矩阵每个元素对应的加起来 (用前缀和优化) 得到一个新的 n 行 1 列的矩阵; col[s2] + … + col[t2] 采用同样方法得到一个新的 1 行 n 列的矩阵. 两个新矩阵再相乘就得到一个只有一个元素的矩阵了, 该元素就是最终答案.

代码

11809ms 49MB

;const int maxn = 2000 + 10;typedef int Matrix[maxn][maxn];LL;Matrix A, B;int main() {int n, m;scanf(“%d %d”, &n, &m);for(int x = 1; x <= n; x++)for(int y = 1; y <= n; y++) {scanf(“%d”, &A[x][y]);A[x][y] += A[x-1][y];} // 前 x 行元素和for(int x = 1; x <= n; x++)for(int y = 1; y <= n; y++) {scanf(“%d”, &B[x][y]);B[x][y] += B[x][y-1];} // 前 y 列元素和for(int i = 0; i < m; i++) {int x1, y1, x2, y2;scanf(“%d %d %d %d”, &x1, &y1, &x2, &y2);if(x1 > x2) swap(x1, x2);if(y1 > y2) swap(y1, y2);LL ans = 0;for(int i = 1; i <= n; i++)ans += (LL)(A[x2][i] – A[x1-1][i]) * (B[i][y2] – B[i][y1-1]);printf(“%lld\n”, ans);}return 0;}

主页

，做事的能力往往只能给你一种机会，