最大子矩阵问题:
问题描述:(具体见http://acm.pku.edu.cn/JudgeOnline/showproblem?problem_id=1050 )
给定一个n*n(0<n<=100)的矩阵,请找到此矩阵的一个子矩阵,并且此子矩阵的各个元素的和最大,输出这个最大的值。
Example:
0 -2 -7 0
9 2 -6 2
-4 1 -4 1
-1 8 0 -2
其中左上角的子矩阵:
9 2
-4 1
-1 8
此子矩阵的值为9+2+(-4)+1+(-1)+8=15。
我们首先想到的方法就是穷举一个矩阵的所有子矩阵,然而一个n*n的矩阵的子矩阵的个数当n比较大时时一个很大的数字 O(n^2*n^2),显然此方法不可行。
怎么使得问题的复杂度降低呢?对了,相信大家应该知道了,用动态规划。对于此题,怎么使用动态规划呢?
让我们先来看另外的一个问题(最大子段和问题):
给定一个长度为n的一维数组a,请找出此数组的一个子数组,使得此子数组的和sum=a[i]+a[i+1]+……+a[j]最大,其中i>=0,i<n,j>=i,j<n,例如
31 -41 59 26 -53 58 97 -93 -23 84
子矩阵59+26-53+58+97=187为所求的最大子数组。
第一种方法-直接穷举法:
maxsofar=0;
for i = 0 to n
{
for j = i to n
{
sum=0;
for k=i to j
sum+=a[k]
if (maxsofar>sum)
maxsofar=sum;
}
}
第二种方法-带记忆的递推法:
cumarr[0]=a[0]
for i=1 to n //首先生成一些部分和
{
cumarr[i]=cumarr[i-1]+a[i];
}
maxsofar=0
for i=0 to n
{
for j=i to n //下面通过已有的和递推
{
sum=cumarr[j]-cumarr[i-1]
if(sum>maxsofar)
maxsofar=sum
}
}
显然第二种方法比第一种方法有所改进,时间复杂度为O(n*n)。
下面我们来分析一下最大子段和的子结构,令b[j]表示从a[0]~a[j]的最大子段和,b[j]的当前值只有两种情况,(1) 最大子段一直连续到a[j] (2) 以a[j]为起点的子段,不知有没有读者注意到还有一种情况,那就是最大字段没有包含a[j],如果没有包含a[j]的话,那么在算b[j]之前的时候我们已经算出来了,注意我们只是算到位置为j的地方,所以最大子断在a[j]后面的情况我们可以暂时不考虑。
由此我们得出b[j]的状态转移方程为:b[j]=max{b[j-1]+a[j],a[j]},
所求的最大子断和为max{b[j],0<=j<n}。进一步我们可以将b[]数组用一个变量代替。
得出的算法如下:
int maxSubArray(int n,int a[])
{
int b=0,sum=-10000000;
for(int i=0;i<n;i++)
{
if(b>0) b+=a[i];
else b=a[i];
if(b>sum) sum=b;
}
return sum;
}
这就是第三种方法-动态规划。
现在回到我们的最初的最大子矩阵的问题,这个问题与上面所提到的最大子段有什么联系呢?
假设最大子矩阵的结果为从第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)
由此我们可以看出最后所求的就是此一维数组的最大子段和问题,到此我们已经将问题转化为上面的已经解决了的问题了。
#include <iostream>
using namespace std;
int ** a;
int **sum;
int max_array(int *a,int n)
{
int *c = new int [n];
int i =0;
c[0] = a[0];
for(i=1;i<n;i++)
{
if(c[i-1]<0)
c[i] = a[i];
else
c[i] = c[i-1]+a[i];
}
int max_sum = -65536;
for(i=0;i<n;i++)
if(c[i]>max_sum)
max_sum = c[i];
delete []c;
return max_sum;
}
int max_matrix(int n)
{
int i =0;
int j = 0;
int max_sum = -65535;
int * b = new int [n];
for(i=0;i<n;i++)
{
for(j=0;j<n;j++)
b[j]= 0;
for(j=i;j<n;j++)//把数组从第i行到第j行相加起来保存在b中,在加时,自底向上,首先计算行间隔(j-i)等于1的情况,然后计算j-i等于 的情况,一次类推,在小间隔的基础上一次累加,避免重复计算
{
for(int k =0;k<=n;k++)
b[k] += a[j][k];
int sum = max_array(b,n);
if(sum > max_sum)
max_sum = sum;
}
}
delete []b;
return max_sum;
}
int main(void)
{
int n;
cin >> n;
a = new int *[n];
sum = new int *[n];
int i =0;
int j =0;
for(i=0;i<n;i++)
{
sum[i] = new int[n];
a[i] = new int[n];
for(j=0;j<n;j++)
{
cin>>a[i][j];
sum[i][j] =0 ;//sum[r][k]表示起始和结尾横坐标分别为r,k时的最大子矩阵
//sum[r][k] = max{sum (a[i][j]):r<=i<=k}:0<=k<=n-1
}
}
/**//*
int b[10]={31,-41,59,26,-53,58,97,-93,-23,84};
cout << max_array(b,10) << endl;
*/
cout << max_matrix(n)<<endl;
return 0;
}
#include <iostream>
#include <string>
const int mt = 101;
int main(void)
{
int a[mt][mt];
int st[mt][mt];
int p,k,n,i,j,sum,maxn;
scanf("%d",&n);
for (i=1;i<=n;i++)
for (j=1;j<=n;j++)
scanf("%d",&a[i][j]);
memset(st,0,sizeof(st));
for (i=1;i<=n;i++)
for (j=1;j<=n;j++)
st[i][j]=st[i][j-1]+a[j][i];
maxn=0;
for (i=1;i<=n;i++)
{
for (j=i;j<=n;j++)
{
p=st[1][j]-st[1][i-1];
sum=p;
for (k=2;k<=n;k++)
{
if (sum>0)
sum+=st[k][j]-st[k][i-1];
else
sum=st[k][j]-st[k][i-1];
if (sum>p)
p=sum;
}
if (p>maxn) maxn=p;
}
}
printf("%d\n",maxn);
return 0;
} code:
#include <iostream>
#include <cstdio>
#include <memory.h>
using namespace std;
int main()
{
// freopen("data.txt","r",stdin);
int arr[110][110],arr2[110],dp[110],n,max;
memset(arr,0,sizeof(arr));
while(cin >> n)
{
max = -2147483647;
for(int i = 1;i <= n;++i)
for(int j = 1;j <= n;++j)
cin >> arr[i][j];
//动态规划过程,将二维数组的和加起来,转化为一维,再利用求最大连续子序列的方法DP求之
for(int i = 1;i <= n; ++i)
{
memset(arr2,0,sizeof(arr2));
for(int j = i; j <= n; ++j) //将矩阵从第i行加到第n行,每加一次算一次最大连续子序列一次
{
memset(dp,0,sizeof(dp));//初始化DP很重要
for(int x = 1;x <= n; ++x)
{
arr2[x] += arr[j][x];
if(dp[x - 1] + arr2[x] > 0) //最大连续子段和的动态转移方程:
{ //DP(i) = 0 if(DP(i-1) + F(i) < 0)
dp[x] = dp[x - 1] + arr2[x];// = DP(i - 1) + F(i) if(DP(i-1) + F(i) > 0)
if(dp[x] > max)
max = dp[x];
}
else
{
dp[x] = 0;
if(dp[x] > max)
max = dp[x];
}
}
}
}
cout << max <<endl;
}
return 0;
}
|