POJ 1050 To the Max 最大子矩阵 HDU1003 最大子段和 dp

题意: 给出一个矩阵, 求子矩阵中元素的和的最大值.

思路:

考虑一维的情况:

一个数列, 求其子段的和的最大值(HDU1003). 这是一个很显然的dp问题.

如果0~i-1的子段和>0, 0~i的子段和就是0~i-1的子段和+a[i] (继承之前的)

不然0~i的子段和就是a[i] (从i开始)

dp[i]=max(dp[i-1]+s[i],s[i]) 

ans=max(dp[0~n])

可以发现, dp可由一个变量代替.

for(i from 1 to len){

      dp=max(dp+s[i],s[i]);

      ans=max(ans,dp);

}

这样, 我们就在O(n)复杂度下解决了求最大子段和的问题.

对于一个N*N的矩阵, 可以枚举列数为N的所有子矩阵, 设这个矩阵的行数为x, 将每个矩阵每一列压缩(相加)为一个数, 就得到一个1行*N列的数列, 通过计算这个数列的最大子段和, 就可以得到这个x*N矩阵的(x*1, x*2, x*3,...,, x*N)的子矩阵的最大和.

参考

假设最大子矩阵的结果为从第r行到k行、从第i列到j列的子矩阵，如下所示(ari表示a[r][i],假设数组下标从1开始)：
  | a11 …… a1i ……a1j ……a1n |
  | a21 …… a2i ……a2j ……a2n |
  |  .     .     .    .    .     .    .   |
  |  .     .     .    .    .     .    .   |
  | ar1 …… ari ……arj ……arn |
  |  .     .     .    .    .     .    .   |
  |  .     .     .    .    .     .    .   |
  | ak1 …… aki ……akj ……akn |
  |  .     .     .    .    .     .    .   |
  | an1 …… ani ……anj ……ann |
那么我们将从第r行到第k行的每一行中相同列的加起来，可以得到一个一维数组如下：
(ar1+……+ak1, ar2+……+ak2, ……,arn+……+akn)
由此我们可以看出最后所求的就是此一维数组的最大子段和问题

将最大子矩阵问题分解为最大子段和。时间复杂度O(n^3)。

 代码:

#include<bits/stdc++.h>

const int M = 105;
int a[M][M];
int n;
int s[M];

int maxSumOfSubArray(int len) {
    int bf = 0;
    int maxX = s[0];
    for (int i = 0; i < len; ++i) {
        bf = max(bf + s[i], s[i]);
        maxX = max(maxX, bf);
    }
    return maxX;
}


void init() {
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            scanf("%d", &a[i][j]);
        }
    }

}

void solve() {
    int ans = a[0][0];
    for (int i = 0; i < n; ++i) {
        memset(s, 0, sizeof(int) * n);
        for (int j = i; j < n; ++j) {
            //每次向s中加一行
            //for完j就是将第i行~第i+1行...第i行~第j行压缩进s中.
            for (int k = 0; k < n; ++k) {
                s[k] += a[j][k];
            }
            ans = max(ans, maxSumOfSubArray(n));
        }
    }
    printf("%d\n", ans);
}

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