题目描述
给定一个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个元素结尾的最大子序和):
d p [ i ] = m a x ( d p [ i − 1 ] , 0 ) + a r r [ i − 1 ] dp[i]=max(dp[i-1],0)+arr[i-1] dp[i]=max(dp[i−1],0)+arr[i−1]
- 一维最大子序和
m a x ( d p [ i ] ) max(dp[i]) max(dp[i])
- 二维的情况与一维类似,只需要遍历第i行到第j行(1<=i<=j<=n)的子矩阵,将子矩阵按列相加,压缩成长度为m的一维子序列,利用一维dp方程,就可以求出以压缩后的最大子序和。
遍历结束后,就可以记录到以 a r r [ i − 1 ] [ k − 1 ] arr[i-1][k-1] arr[i−1][k−1]到 a r r [ j − 1 ] [ k − 1 ] arr[j-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 ) O(mn^2) O(mn2)