一个M*N的矩阵,找到此矩阵的一个子矩阵,并且这个子矩阵的元素的和是最大的,输出这个最大的值。
例如:3*3的矩阵:
-1 3 -1
2 -1 3
-3 1 2
和最大的子矩阵是:
3 -1
-1 3
1 2
Input
第1行:M和N,中间用空格隔开(2 <= M,N <= 500)。
第2 - N + 1行:矩阵中的元素,每行M个数,中间用空格隔开。(-10^9 <= Mii <= 10^9)
Output
输出和的最大值。如果所有数都是负数,就输出0。
Sample Input
3 3 -1 3 -1 2 -1 3 -3 1 2
Sample Output
7
解题思路:使用一个前缀dp数组保存从第i行到第j行的每一列的元素和保存下来,然后按一维求最大子序列和:(相当于把二维压缩成一维)
AC代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn=550;
int main()
{
int dp[maxn][maxn],m,n,x;
cin>>m>>n;
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
cin>>x;
dp[i][j]=dp[i-1][j]+x;
}
int ans=0;
for(int i=1;i<=n;i++)
for(int j=i;j<=n;j++)
{
int sum=0;
for(int k=1;k<=m;k++)
{
sum+=dp[j][k]-dp[i-1][k];
if(sum<0)sum=0;
if(sum>ans) ans=sum;
}
}
cout<<ans<<endl;
return 0;
}