题目描述
给定一个n*m的矩阵A,求A中的一个非空子矩阵,使这个子矩阵中的元素和最大。
其中,A的子矩阵指在A中行和列均连续的一块。
样例说明
取最后一列,和为10。
数据规模和约定
对于100%的数据,1< =n, m< =500,A中每个元素的绝对值不超过5000。
输入
输入的第一行包含两个整数n, m,分别表示矩阵A的行数和列数。
接下来n行,每行m个整数,表示矩阵A。
输出
输出一行,包含一个整数,表示A中最大的子矩阵中的元素和。
样例输入
3 3
-1 -4 3
3 4 -1
-5 -2 8
样例输出
10
这道题其实是二维的最大字段和的问题,我们只需要把上面的每一列压缩下来,枚举每一行,二维变一维就好做了。
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const int N=510;
int a[N][N];
int sc[N];
int n,m;
int get_ans(int a[])
{
int ans=0;
int maxv=-1e9;
int dp[N];
dp[0]=a[0];
for(int i=1;i<m;i++)
{
dp[i]=max(a[i],a[i]+dp[i-1]);
maxv=max(dp[i],maxv);
}
return maxv;
}
int main()
{
cin>>n>>m;
int ans=-1e9;
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
cin>>a[i][j];
for(int i=0;i<n;i++)
{
memset(sc,0,sizeof sc);
for(int j=i;j>=0;j--)
{
for(int k=0;k<m;k++)
{
sc[k]+=a[j][k];
}
ans=max(ans,get_ans(sc));
}
}
cout<<ans<<endl;
return 0;
}