一个M*N的矩阵,找到此矩阵的一个子矩阵,并且这个子矩阵的元素的和是最大的,输出这个最大的值。
例如:3*3的矩阵:
-1 3 -1
2 -1 3
-3 1 2
和最大的子矩阵是:
3 -1
-1 3
1 2
输入
第1行:M和N,中间用空格隔开(2 <= M,N <= 500)。 第2 - N + 1行:矩阵中的元素,每行M个数,中间用空格隔开。(-10^9 <= M[i] <= 10^9)
输出
输出和的最大值。如果所有数都是负数,就输出0。
输入示例
3 3 -1 3 -1 2 -1 3 -3 1 2
输出示例
7
解题思路:最大子矩阵一定在某两行之间,假设最大矩阵位于第i行到第j行之间,这时我们只需要判断最大子矩阵在哪两列之间就行了,枚举复杂度太高,所以我们可以将二维的矩阵压缩成一维,用数组c[]保存每一列第i行到第j行之间的和 ,然后求数组c的最大连续子段和,就可以了
AC代码:
#include<iostream>
using namespace std;
const int maxn=1e3;
int c[maxn];
int a[maxn][maxn];
int solve(int b[],int n)//最大子段和
{
int endmax,answer;
endmax=answer=b[0];
for(int i=1;i<n;i++)
{
endmax=max(endmax,0)+b[i];
answer=max(endmax,answer);
}
return answer;
}
int main()
{
int m,n,ans=0,i,j,k;
cin>>m>>n;
for(i=0;i<n;i++)
{
for(j=0;j<m;j++)
cin>>a[i][j];
}
for(i=0;i<n;i++)
{
for(j=i;j<n;j++)
{
for(k=0;k<m;k++)
{
c[k]=(j==i)?a[i][k]:(c[k]+a[j][k]);//每一列第i行到第j行之间的和
}
ans=max(ans,solve(c,m));
}
}
cout<<ans<<endl;
return 0;
}