合工大 程序设计艺术 实验四 最大子矩阵和

题目

输入一个矩阵，计算所有的子矩阵中和的最大值。
例如，输入
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
输出为：15
思考：当矩阵很大时，比如 100*100 的矩阵，你的程序还能够很快的得出结果吗，如果
不能，请思考如何用动态规划的思想解决。

说明

这题类似于求最大字段和，不同的是从一维上升到了二维。

解题思路

这题我用到了前缀和的思想。
我用a[N][N]储存要求解的矩阵，s[N][N]储存前缀和。
以这张图为例，i=3,j=3的前缀和s[3][3]=s[2][3]+s[3][2]-s[2][2]+a[3][3];
所以有s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
利用maxx记录当前最大值，遍历求解出x1,x2,y1,y2围成的矩阵和，该和于maxx作比较，若更大则更新最大值。
枚举结束后，最后的maxx就是我们要求解的最大子矩阵和。
运行结果：

思考

我这个算法的时间复杂度是O（n^4）
并且我在中间加了一个判断
if (x2 < x1 || y2 < y1) {continue;}这样可以避免一些无意义的计算，若左上角比右下角大，不用求解。
如图所示当n=100时，程序运算出结果用时1.941s。还是可以很快得出结果的。

源代码

#include <iostream>
#include <algorithm>
#include <ctime>
#define N 125
using namespace std;
int a[N][N];//要求值的矩阵
int s[N][N];//前缀和
int main() {
	int n;
	cin >> n;//n*n的矩阵
	//clock_t start, finish;
	//double duration;
	//start = clock();
	//输入矩阵数据
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			//int b = rand() % 254 - 127;
			//a[i][j] = b;
			//cout << b << '\t';
			cin >> a[i][j];
		}
		cout << endl;
	}
	//前缀和
	for (int i = 1; i <= n; i++) {
		for (int j = 1; j <= n; j++) {
			s[i][j] = s[i - 1][j] + s[i][j - 1] - s[i - 1][j - 1] + a[i][j];
		}
	}
	int maxx = -99999;//设置最大值
	int sum = 0;
	for (int x1 = 1; x1 <= n; x1++) {
		for (int y1 = 1; y1 <= n; y1++) {
			for (int x2 = 1; x2 <= n; x2++) {
				for (int y2 = 1; y2 <= n; y2++) {
					//若左上角比右下角大，不用求解
					if (x2 < x1 || y2 < y1) {
						continue;
					}
					//求x1,x2,y1,y2区域的矩阵和
					sum = s[x2][y2] - s[x2][y1 - 1] - s[x1 - 1][y2] + s[x1 - 1][y1 - 1];
					maxx = max(maxx, sum);//更新最大值
				}
			}
		}
	}
	//finish = clock();
	//duration = (double)(finish - start) / CLOCKS_PER_SEC;
	//printf("Time is %lf\n", duration);
	cout << maxx;
	return 0;
}
//时间复杂度O（n^4）