51nod 1051 最大子矩阵和

1051 最大子矩阵和

一个M*N的矩阵，找到此矩阵的一个子矩阵，并且这个子矩阵的元素的和是最大的，输出这个最大的值。

输入

第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

解法：

设最后形成的答案子矩阵是 【 (x1, y1), ( x2, y2 )  】, 那么< x1, x2 > 的组合共有n^2个，枚举n^2个组合，每个组合用行上的前缀和求出 （  x1，k ）、( x2，k ) 上数字的和，然后 O(m)复杂度求最大子段和。总复杂度 O(n^2 * m).

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long Lint;
const int max_n = 1e5+1e2;

int n,m;
Lint num[510][510];
Lint sum[510][510];
Lint a[510];

Lint GetMax(){
    Lint res=-1e15;
    for(int i=1;i<=m;i++){
        a[i]=max(a[i],a[i]+a[i-1]);
        res=max(res,a[i]);
    }
    return res;
}

void solve(){
    scanf("%d %d",&m,&n);
    Lint res=-1e15;
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            scanf("%lld",&num[i][j]);
            sum[i][j]=num[i][j]+sum[i-1][j];
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=i;j<=n;j++){
            for(int k=1;k<=m;k++)
                a[k]=sum[j][k]-sum[i-1][k];
            res=max(res,GetMax());
        }
    }
    printf("%lld\n",res);
}

int main(){
    solve();
    return 0;
}