描述
一个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
思路
枚举子矩阵的第一行,固定第一行,枚举矩阵的最后一行。
每次记录当前子矩阵的列元素的和,压缩状态求最大子段和。
要注意,初值应赋int的最小值,而不是0;注意m,和n的输入顺序。
AC代码
#include<bits/stdc++.h>
using namespace std;
int m,n;
int a[510][510];
int sum[510];
int max1,ans;
int maxsum(int aa[],int n)
{
int msum = 0x80000000, b = 0;
for (int i = 0; i < n; i++)
{
if (b > 0)
b += aa[i]; ///前面的连续子段之和大于0就累加
else
b = aa[i]; ///前面的连续子段之和小于等于0就重置b
msum = max(msum, b);
}
return msum;
}
int main()
{
scanf("%d %d",&m,&n);
for(int i=0; i<n; i++)
for(int j=0; j<m; j++)
scanf("%d",&a[i][j]);
ans=0x80000000;///初始化为最小值
for(int i=0; i<n; i++)///枚举第一行,上界
{
memset(sum,0,sizeof(sum));///列的和,每次都要重置
for(int j=i; j<n; j++)///枚举最后一行,下届
{
for(int k=0; k<m; k++)
sum[k]+=a[j][k];//求每列的和
max1=maxsum(sum,m);
ans=max(max1,ans);
}
}
printf("%lld\n",ans);
}