题意: 给出一个矩阵, 求子矩阵中元素的和的最大值.
思路:
考虑一维的情况:
一个数列, 求其子段的和的最大值(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;
}