【编程之美】2.15 子数组之和的最大值(二维)
思路,有了一维的思路,我们想办法把二维问题转化为一维的问题。
我们假定已经选中了行的范围是 a-c 那么把每一列中 a-c的元素加起来就变成了一个一维的问题。只需对行的范围遍历,再用一维的方法来解就可以了。
注意,也可以对列的范围遍历,行和列那个小就对哪个遍历。 复杂度为O(M * N * min(M, N))
#include <stdio.h> #include <stdlib.h> void getSumCol(int * a, int aColNum, int RowBegin, int RowEnd, int * SumColArray) { for(int j = 0; j < aColNum; j++) { SumColArray[j] = 0; for(int i = RowBegin; i <= RowEnd; i++) { SumColArray[j] += *(a + i * aColNum + j); } } } int getMax2DArraySum(int * a,const int aRowNum,const int aColNum) { int MaxSum = a[0]; int Sum = 0; int * SumColArray = (int *)calloc(aColNum, sizeof(a[0])); for(int i = 0; i < aRowNum; i++) { for(int k = i; k < aRowNum; k++) { getSumCol(a, aColNum, i, k, SumColArray); Sum = SumColArray[0]; for(int j = 1; j < aColNum; j++) { Sum = (Sum > 0) ? Sum + SumColArray[j] : SumColArray[j]; MaxSum = (Sum > MaxSum) ? Sum : MaxSum; } } } free(SumColArray); return MaxSum; } int main() { int a[4][5] = { { 1, 2, 3, 4, 5}, {-3,-5, 3,-8, 0}, { 2, 3, 5, 6, 8}, { -3,1, 2, 3, 0} }; int max = getMax2DArraySum((int*)a, 4, 5); return 0; }