基準時間限制:2 秒 空間限制:131072 KB 分值: 40 難度:4級算法題
一個M*N的矩陣,找到此矩陣的一個子矩陣,並且這個子矩陣的元素的和是最大的,輸出這個最大的值。
例如:3*3的矩陣:
-1 3 -1
2 -1 3
-3 1 2
和最大的子矩陣是:
3 -1
-1 3
1 2
Input
第1行:M和N,中間用空格隔開(2 <= M,N <= 500)。 第2 - N + 1行:矩陣中的元素,每行M個數,中間用空格隔開。(-10^9 <= M[i] <= 10^9)
Output
輸出和的最大值。如果所有數都是負數,就輸出0。
Input示例
3 3 -1 3 -1 2 -1 3 -3 1 2
Output示例
7
#include<iostream>
#include<cstring>
typedef long long ll;
using namespace std;
ll sum,a[520][520],dp[520];
ll N,M,mas=-1,temp;
ll maxsub()
{
ll mas=0,t=0;
for(int i=0;i<M;i++)
{
if(t>0)t+=dp[i];
else t=dp[i];
if(t>mas)mas=t;
}
return mas;
}
int main()
{
cin>>M>>N;
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(dp,0,sizeof(dp));
for(int j=i;j<N;j++)
{
for(int k=0;k<M;k++)
dp[k]+=a[j][k];
temp=maxsub();
if(temp>mas)mas=temp;
}
}
if(mas>=0)cout<<mas<<endl;
else cout<<0<<endl;
return 0;
}