POJ 1050 最大子矩阵和

来源:互联网 发布:client mac and addr 编辑:程序博客网 时间:2024/05/16 15:35

POJ 1050 最大子矩阵和

题目:http://poj.org/problem?id=1050
题意:给一个N*N的矩阵,求一个子矩阵,使得该子矩阵中的数字之和为所有子矩阵中最大的。
思路:这里给出了两种思路:
1.枚举所有可能的组合。
2.dp,其实这是根据一维的最大子段和改编的题目

枚举:
首先计算出所有的A[0][0]至A[i][j]这样的子矩阵的和,当然不是一个个的计算,有一点点小的技巧,具体参见代码。
然后利用四重循环枚举所有可能的矩阵的左上角顶点(i,j)和右下角顶点(ii,jj),则这个子矩阵的和即为A[ii][jj]-A[i-1][jj]-A[ii][j-1]+A[i-1][j-1].这样找出所有的和中最大的值即为所求。
代码:

#include <cstdio>#include <cstring>#include <iostream>using namespace std;const int maxn=107;int a[maxn][maxn];int main(){    int n;    scanf("%d",&n);    memset(a,0,sizeof(a));    for(int i=1;i<=n;i++)    for(int j=1;j<=n;j++)    {        scanf("%d",&a[i][j]);        if(i==0) a[i][j]+=a[i][j-1];        else if(j==0) a[i][j]+=a[i-1][j];        else a[i][j]+=(a[i-1][j]+a[i][j-1]-a[i-1][j-1]);    }    //枚举所有可能    int ans=a[1][1];    for(int i=1;i<=n;i++)    for(int j=1;j<=n;j++)    for(int ii=i;ii<=n;ii++)    for(int jj=j;jj<=n;jj++)        if(ii==i&&jj==j) continue;        else ans=max(ans,a[ii][jj]-a[i-1][jj]-a[ii][j-1]+a[i-1][j-1]);    printf("%d\n",ans);}

dp思路:
其实就是每次选择两行i,j,并将第i至第j行间的元素全部压缩至一行中,这样问题就变成了一维数组的最大子段和问题,其状态转移方程为
p[i]=p[i-1]>0? p[i-1]+p[i]:p[i];
注意要枚举所有合理的i,j组合,故每次选择的i和j的顺序十分重要,此代码中的i,j枚举顺序可以避免重复运算,减少时间开销。
代码:

#include <cstring>#include <iostream>using namespace std;const int maxn=107;int a[maxn][maxn];int main(){    int n;    scanf("%d",&n);    int ans=-128;    int tmp;    for(int i=1;i<=n;i++)    {        tmp=0;        for(int j=1;j<=n;j++)        {            scanf("%d",a[i]+j);            if(tmp<0) tmp=a[i][j];            else tmp+=a[i][j];            if(tmp>ans)                ans=tmp;        }    }    for(int i=1;i<n;i++)            //每次选择i,j两行之间的合并为一行,注意i,j选择的顺序很重要,这        for(int j=i+1;j<=n;j++)     //样选择可以尽量减少运算        {            tmp=0;            for(int k=1;k<=n;k++)            {                a[i][k]+=a[j][k];                if(tmp<0) tmp=a[i][k];                else tmp+=a[i][k];                if(tmp>ans) ans=tmp;            }        }    printf("%d\n",ans);}
1 0
原创粉丝点击