问题描述
给定一个n*m的矩阵A,求A中的一个非空子矩阵,使这个子矩阵中的元素和最大。
其中,A的子矩阵指在A中行和列均连续的一块。
输入格式
输入的第一行包含两个整数n, m,分别表示矩阵A的行数和列数。
接下来n行,每行m个整数,表示矩阵A。
输出格式
输出一行,包含一个整数,表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
样例说明
取最后一列,和为10。
数据规模和约定
对于50%的数据,1<=n, m<=50;
对于100%的数据,1<=n, m<=500,A中每个元素的绝对值不超过5000。
步骤一,预处理,每个数存的是从第一行当前行的每行的当前的列的值的和。每次想要的到当前列值和或者几行的列值和,通过相减就好。这是纵向的压缩,然后进行横向的压缩。第三个循环r 从1开始到n,就像一维一样,进行压缩。求得当前矩阵k,如果sum>0 sum+=k 如果sum<=0 sum=k; 然后每次比较记录最大的子阵值,但是还有一种情况是全部都是负数,所以还要加一个条件,如果全部都是负数,那么输出最大的负数。
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
int zhi[1000][1000];
int dp[1000][1000];
int main()
{
int n,m;
cin>>n>>m;
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
{
int a;
cin>>a;
zhi[i][j]=zhi[i-1][j]+a;
}
int sum=0;
int maxi=-5005;
int maxm=-5005;
int flag=0;
for(int i=0;i<n;i++)
{
for(int j=i+1;j<=n;j++)
{
sum=-1;
for(int r=1;r<=m;r++)
{
int k=zhi[j][r]-zhi[i][r];
if(sum>0)
sum+=k;
else
{
sum=k;
if(!flag&&maxi<sum) maxi=sum,flag=1;
}
if(sum>maxm)
{
maxm=sum;
flag=1;
}
}
}
}
if(!flag)
maxm=maxi;
printf("%d\n",maxm );
}