问题:
求一个M*N的矩阵的最大子矩阵和。
比如在如下这个矩阵中:
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2 ………………【1】
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
拥有最大和的子矩阵为:
9 2
-4 1
-1 8
其和为15。
-4 1
-1 8
其和为15。
分析:
我们首先想到的方法就是穷举一个矩阵的所有子矩阵,然而一个n*n的矩阵的子矩阵的个数当n比较大时时一个很大的数字 O(n^2*n^2),显然此方法不可行。
怎么使得问题的复杂度降低呢?对了,相信大家应该知道了,用动态规划。
思路:假设当前考虑从第r行到第k行的矩阵,可以先求从第r行到第k行每一列的元素和,放在一维数组中。那么现在就转换成求一维数组的最大字段和问题了!
循环DP,共三层循环。
最外层i循环1-N,表明子矩阵是从第i列开始累加的。
第二层j循环i-N,表明子矩阵是从第i列累加到第j列。
第三层k从1到M做一维DP。
//Attention: a does not stand for two_dimensional array's first address!
//It actually stand for two_level point!!
int MaxSum(int row,int col,int **a)
{
int i,j,k,sum=0,max;
int *b=(int *)malloc(sizeof(int)*col+1);
for (i=1;i<=row;i++)
{
for (k=1;k<=col;k++)
{
b[k]=0;
}
for (j=i;j<=row;j++)
{
for (k=1;k<=col;k++)
{
b[k]+=a[j][k];
}
max=MaxSequence(col,b);
if (max>sum)
{
sum=max;
}
}
}
return sum;
}