最大子阵——动态规划DP

历届试题 最大子阵  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。

　　其中，A的子矩阵指在A中行和列均连续的一块。

输入格式

　　输入的第一行包含两个整数n, m，分别表示矩阵A的行数和列数。
　　接下来n行，每行m个整数，表示矩阵A。

输出格式

　　输出一行，包含一个整数，表示A中最大的子矩阵中的元素和。

样例输入

3 3
-1 -4 3
3 4 -1
-5 -2 8

样例输出

10

样例说明

　　取最后一列，和为10。

数据规模和约定

　　对于50%的数据，1<=n, m<=50；
　　对于100%的数据，1<=n, m<=500，A中每个元素的绝对值不超过5000。

-1	-4	3
3	4	-1
-5	-2	8

-1	-4	3
2	0	2
-3	-2	10

1、相当于对一维的最大和的拓展。

2、记录每一列的sum

3、dp[j][r] - dp[i-1][r]就是对 i~j 行的数值的求和

4、接下来的，相当于对每一行进行最大和求解，也就是对数组进行最大和求解

5、注意全是负数的情况

#include<iostream>
#include<stdio.h>
using namespace std;
int dp[1005][1005];
int main()
{
    int n,m;
    int k;
    cin>>n>>m;
    for(int j = 0 ; j < m; j++)
    {
        dp[0][j] = 0;
    }
    for(int i = 0; i < n; i ++)
    {
        for(int j = 0 ; j< m ;j ++)
        {
            scanf("%d",&k);
            dp[i][j] = dp[i-1][j] + k;
        }
    }
    int sum = 0;
    int maxx = 0;
    int flag = 0;
    int maxi = -100005;
    for(int i = 0 ; i < n ;i++)
        for(int j = i; j < n ;j++ )
    {
        sum = 0;
        for(int r = 0 ; r < m ;r++)
        {
            if(sum > 0 ) sum += (dp[j][r]-dp[i-1][r]);
            else
            {
                sum = (dp[j][r]-dp[i-1][r]);
                if(!flag && sum > maxi)
                    maxi = sum;
            }
            if(sum > maxx )
            {
                maxx =sum;
                flag = 1;
            }

        }
    }
    if(!flag)
    {
        maxx = maxi;
    }
    cout<<maxx<<endl;
}