问题描述:
给定一个整型矩阵,返回子矩阵的最大累计和。
思想
我们可以在矩阵中首先取出一个子矩阵,然后遍历这个子矩阵的全部数据得出子矩阵的累加和。在一个矩阵画出一个子矩阵的时间复杂度为O(N^4),遍历子矩阵获得累加和时间复杂度为O(N^2),所以该思想的时间复杂度为O(N^6)。
我们可以对算法进行优化。
假设数组arr[3][3],我们依次查找0-0, 0-1, 0-2,1-1, 1-2, 2-2行的子矩阵最大值,取其中的最大值输出。在查找0-1行的子矩阵最大值的流程为:把第一行和第零行的数字进行相加,查找数组中的最大累加和(可以参考:给定数组返回子数组的最大累加和_朂後 哋箹萣的博客-CSDN博客)。该算法的时间复杂度可以从O(N^6)降到O(N^3)。
若输入的数组arr[n][m](n<m)时,算法的时间复杂度可以变为:O(N^2*M)。
代码
public static int maxSum(int[][] m) {
if (m == null || m.length == 0 || m[0].length == 0) {
return 0;
}
int max = Integer.MIN_VALUE;
int cur = 0;
int[] s = null;
for (int i = 0; i != m.length; i++) {
s = new int[m[0].length];
for (int j = i; j != m.length; j++) {
cur = 0;
for (int k = 0; k != s.length; k++) {
s[k] += m[j][k];
cur += s[k];
max = Math.max(max, cur);
cur = cur < 0 ? 0 : cur;
}
}
}
return max;
}