二维最大子序列和

题目描述

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

• 数据规模和约定
  对于50%的数据，1<=n, m<=50；
  　　对于100%的数据，1<=n, m<=500，A中每个元素的绝对值不超过5000。
• 输入
  　　输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
  　　接下来n行，每行m个整数，表示矩阵A。
• 输出
  输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。
• 样例输入

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

• 样例输出

10

• 样例说明
  取最后一列，和为10。

解题思路

• 回忆：一维最大子序列和的dp方程（dp[i]表示以第i个元素结尾的最大子序和）：

$dp[i]=max(dp[i-1],0)+arr[i-1]$

• 一维最大子序和

$max(dp[i])$

• 二维的情况与一维类似，只需要遍历第i行到第j行（1<=i<=j<=n）的子矩阵，将子矩阵按列相加，压缩成长度为m的一维子序列，利用一维dp方程，就可以求出以压缩后的最大子序和。
  遍历结束后，就可以记录到以$arr[i-1][k-1]$到$arr[j-1][k-1]$为最后一列的子矩阵的最大子序和（1<=i<=j<=n，1<=k<=m），用max记录最大值即可。

C++代码

#include<bits/stdc++.h>
using namespace std;
int n, m;
int main() {
	cin >> n >> m;
	vector < vector<int>>arr(n,vector<int>(m));
	for (int i = 0; i < n; i++)
	{
		for (int j = 0; j < m; j++)
		{
			cin >> arr[i][j];
		}
	}
	//dp[n][m]表示矩阵第n列前m行的元素之和
	vector<vector<int>>dp(n+1, vector<int>(m+1));
	for (int i = 1; i <= n; i++)
	{
		for (int j = 1; j <= m; j++)
		{
			dp[i][j] = dp[i - 1][j] + arr[i - 1][j - 1];
		}
	}
	
	int max = -5000;
	for (int i = 1; i <= n; i++)
	{
		for (int j = i; j <= n; j++)
		{
			int sum = 0;// sum记录当前压缩序列中 以第k-1个元素结尾的最大子序和
			for (int k = 1; k <= m; k++)
			{
				//dp[j][k] - dp[i - 1][k]即为将第i-j行子矩阵压缩成一维序列后的第k个数
				if (sum >= 0)
				{
					sum += dp[j][k] - dp[i - 1][k];
				}
				else {
					sum = dp[j][k] - dp[i - 1][k];
				}
				max = max > sum ? max : sum;
			}
		}
	}
	cout << max;
	return 0;
}

算法分析

时间复杂度：$O(m{n}^2)$