算法真题03 - 二维的最大连续子序列和

（yitu）面试官先问我一个1维的，leetcode 53. 最大子序和，我很快答上来了，后来拓展成 二维子序列和。我用动态规划，瞎讲了一通，结果没有蒙骗过去，最后转换为时间复杂度比较高的一维问题解决了。

先看一下一维怎么解决。
输入: [-2,1,-3,4,-1,2,1,-5,4],
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大，为 6。

用一维dp，dp[i]表示到第i个数为止，最大的和。那么它的递归为dp[i] = max(dp[i-1],0) + arr[i]
因为我自己arr[i]是一定要考虑的。如果前面的最大和大于0，就加在我自己上，否则为负数还不如不要这个累赘。

#include <iostream>
#include <vector>  
#include<algorithm>
using namespace std;

int maxSubArray(vector<int> arr) {
	int n = arr.size();
	vector<int> dp(n, 0);
	int res = arr[0];
	for (int i = 1; i < n; i++) {
		dp[i] = max(dp[i - 1], 0) + arr[i];
		if (dp[i] > res) res = dp[i];
	}
	return res;
}

int main()
{
	int n;
	cin >> n;
	vector<int> arr;
	int t;
	for (int i = 0; i < n; i++) {
		cin >> t;
		arr.push_back(t);
	}
	cout << maxSubArray(arr) << endl;
}
/*
9
-2 1 -3 4 -1 2 1 -5 4
*/

问题描述
给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。
其中，A的子矩阵指在A中行和列均连续的一块。

输入格式
输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
接下来n行，每行m个整数，表示矩阵A。

输出格式
输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。

样例输入

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

样例输出

10

图解：

#include <iostream>
#include <vector>  
#include<algorithm>
using namespace std;

int main()
{
	int n,m;
	cin >> n >> m;
	vector<vector<int>> dp(n+1,vector<int>(m+1,0));
	int t;
	for (int i = 1; i <= n; i++) { //圈1
		for (int j = 1; j <= m; j++) {
			cin >> t;
			dp[i][j] = dp[i - 1][j] + t;
		}
	}
	int res = -9999;
	for (int i = 1; i <= n; i++) {//圈2
		for (int j = 1; j <= n; j++) {//控制第i行到第j行
			int s = 0,tmp;
			for (int k = 1; k <= m; k++) {
				tmp = dp[j][k] - dp[i - 1][k];//一竖条的面积
				s = max(s, 0) + tmp;//变成了一维
				res = max(res, s);
			}
		}
	}
	cout << res << endl;

}
/*
3 3
-1 -4 3 
3 4 -1 
-5 -2 8
*/

参考资料：这里