问题描述
给定一个n*m的矩阵A,求A中的一个非空子矩阵,使这个子矩阵中的元素和最大。
其中,A的子矩阵指在A中行和列均连续的一块。
其中,A的子矩阵指在A中行和列均连续的一块。
输入格式
输入的第一行包含两个整数n, m,分别表示矩阵A的行数和列数。
接下来n行,每行m个整数,表示矩阵A。
接下来n行,每行m个整数,表示矩阵A。
输出格式
输出一行,包含一个整数,表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
样例说明
取最后一列,和为10。
数据规模和约定
对于50%的数据,1<=n, m<=50;
对于100%的数据,1<=n, m<=500,A中每个元素的绝对值不超过5000。
解题思路:
因为要求所有的行列都是连续的,因此我们枚举所有可能情况就行了,但这里有一个技巧,我们可以提前处理好每一列的前n行的和,然后枚举的时候只需枚举一个行的开始,和行的结束,然后对这个范围的所有列(即m的值)进行最大子段和的dp就OK啦,这一过程中不断维护最大值;
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define maxn 510
int num[maxn][maxn];
int sum[maxn][maxn];
int dp_sum(int *a,int n)
{
int ret=a[1],dp[maxn];
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++){
dp[i]=max(a[i],dp[i-1]+a[i]);
ret=max(dp[i],ret);
}
return ret;
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
memset(sum,0,sizeof(sum));
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
scanf("%d",&num[i][j]);
sum[i][j]=sum[i-1][j]+num[i][j];
}
int maxnum=num[1][1];
for(int i=1;i<=n;i++)
for(int j=i;j<=n;j++)
{
int tem[maxn];
memset(tem,0,sizeof(tem));
for(int k=1;k<=m;k++)
tem[k]=sum[j][k]-sum[i-1][k];
maxnum=max(maxnum,dp_sum(tem,m));
}
printf("%d\n",maxnum);
}