最大子矩阵和问题归纳总结

一,最大子矩阵问题:
给定一个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),显然此方法不可行。怎么使得问题的复杂度降低呢?对了,相信大家应该知道了,用动态规划。对于此题,怎么使用动态规划呢?

    请先参考-->最大子段和问题
    这个问题与最大子段有什么联系呢?

1、首先考虑一维的最大子段和问题,给出一个序列a[0],a[1],a[2]…a[n],求出连续的一段,使其总和最大。

a[i]表示第i个元素
dp[i]表示以a[i]结尾的最大子段和

dp[i] = max{a[i], dp[i-1] + a[i]}

解释一下方程:

如果dp[i-1] > 0,则 dp[i] = dp[i-1] + a[i]
如果dp[i-1] < 0,则 dp[i] = a[i]

因为不用记录位置信息,所以dp[]可以用一个变量dp代替:

如果dp > 0,则dp += a[i]
如果dp < 0,则dp = a[i]

2、考虑二维的最大子矩阵问题

我们可以利用矩阵压缩把二维的问题转化为一维的最大子段和问题。因为是矩阵和,所以我们可以把这个矩形的高压缩成1,用加法就行了。

假设最大子矩阵的结果为从第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)
由此我们可以看出最后所求的就是此一维数组的最大子段和问题,到此我们已经将问题转化为上面的已经解决了的问题了。

poj1050:给一个n*n(1<=n<=100)的矩阵,求最大子矩阵和

#include 
#include 
#include 
#include 
#include 
using namespace std;
int mp[100][100];
int temp[200];
const  int mod = 1e9+7;

int solve(int *a,int n)
{
    int dp = 0,Max = 0;
    for(int i = 0; i < n; i++)
    {
        if(dp > 0) dp += a[i];
        else dp = a[i];

        Max = max(Max, dp);
    }
    return Max;
}

int main()
{
    #ifdef xxz
      freopen("in.txt","r",stdin);
    #endif // xxz
    int n;
    while(~scanf("%d",&n))
    {
       for(int i = 0; i < n; i++)
        for(int j = 0; j < n; j++)
            cin>>mp[i][j];

       int max_ans = 0;

       for(int i = 0; i < n; i++)
       {
           memset(temp,0,sizeof(temp));
           for(int j = i;j < n; j++)
           {
               for(int k = 0; k < n; k++)
                    temp[k] += mp[j][k];

                int ans = solve(temp,n);
                max_ans = max(max_ans,ans);
           }
       }
       cout<return 0;
}

HDU1559:给出n*m(0< n,m<=1000)的矩阵,求规格为x*y的小矩阵最大和为多少。
这题相比上题增加了限制条件,也就是上面那种的特殊情况,如用上面算法时间复杂度是O(n*n*m)则会超时,由于题目对小矩阵有了限制,我们可以用DP做,令a[i][j]为1<=s<=i,1<=t<=j所有元素的和,可以在线处理,那么就可以通过加减来求出当前子矩阵和,就好像计算几个矩阵面积一样(某部分覆盖在一起,所以相加后减去重复的面积)。

#include 
#include 
#include 
#include 
#include 
using namespace std;
int mp[1000][1000];
int temp[200];
const  int mod = 1e9+7;

int solve(int *a,int n)
{
    int dp = 0,Max = 0;
    for(int i = 0; i < n; i++)
    {
        if(dp > 0) dp += a[i];
        else dp = a[i];

        Max = max(Max, dp);
    }
    return Max;
}

int main()
{
#ifdef xxz
    freopen("in.txt","r",stdin);
#endif // xxz
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m,x,y;
        cin>>n>>m>>x>>y;
        int Max = 0;

        for(int i = 1; i <= n; i++)
        {
            for(int j = 1; j <= m; j++)
            {
                cin>>mp[i][j];

                mp[i][j] += mp[i-1][j] + mp[i][j-1] - mp[i-1][j-1];

                if(i >= x && j >= y)
                {
                    int ans = mp[i][j] - mp[i-x][j] - mp[i][j-y] + mp[i-x][j-y];
                    Max = max(Max,ans);
                }
            }
        }


        cout<return 0;
}

其实这题还可以用数状数组做,二维的裸题

#include 
#include 
#include 
using namespace std;

int S;

int mp[1200][1200];
int lowbit(int x)
{
    return x & -x;
}

int  getsum(int x,int y)
{
     int  sum  = 0;
     for(int i  = x; i > 0; i -= lowbit(i))
        for(int j = y; j > 0; j -= lowbit(j))
     {
           sum += mp[i][j];
     }
     return sum;
}

void update(int x,int y,int value)
{
    for(int i = x; i <= 1200; i += lowbit(i))
    for(int j = y; j <=1200 ; j += lowbit(j))
    {
        mp[i][j] += value;
    }
}


int main()
{
    #ifdef xxz
    freopen("in","r",stdin);
    #endif // xxz
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int m,n,x,y;
        scanf("%d%d%d%d",&m,&n,&x,&y);
        memset(mp,0,sizeof(mp));
        for(int i = 1; i <= m; i++)
        {
            for(int j = 1; j <= n; j++)
            {
                int temp;
                scanf("%d",&temp);
                update(i,j,temp);
            }
        }

        int ans = -1;
        for(int i = x; i <= m; i++)
        {
            for(int j = y; j <= n; j++)
            {
                ans = max(ans,getsum(i,j) - getsum(i-x,j)-getsum(i,j-y) + getsum(i-x,j-y));
            }
        }
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(矩阵,线段树,&,树状数组)