VOJ 最大子阵 题解 动态规划

最大子阵

题目描述

给定一个n*m的矩阵A，求A中的一个非空子矩阵，使这个子矩阵中的元素和最大。
其中，A的子矩阵指在A中行和列均连续的一块。

数据规模和约定

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

输入格式

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

输出格式

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

样例

用例输入

3 3
2 -4 1
-1 2 1
4 -2 2

用例输出

6

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int n, m;        // 行，列
int num[56][56]; // 读入的矩阵
int dp[56];      // 将当前子阵所有行合并后，求最大连续子序列
int tmp[56];     // 存储第i列的元素总和
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    cin >> n >> m;
    for (int i = 0; i < n; i++)
    {
        for (int j = 0; j < m; j++)
        {
            cin >> num[i][j];
        }
    }
    int ans = num[0][0];
    for (int i = 0; i < n; i++) // 遍历子阵的起始行
    {
        memset(tmp, 0, sizeof(tmp));
        for (int j = i; j < n; j++) // 遍历子阵的终止行
        {
            for (int k = 0; k < m; k++) // 遍历列数
                tmp[k] += num[j][k];
            memset(dp, 0, sizeof(dp)); // dp用来求当前子阵所有行加起来的最大连续子序列
            dp[0] = tmp[0];
            for (int u = 1; u < m; u++)
            {
                dp[u] = max(dp[u - 1] + tmp[u], tmp[u]);
                ans = max(dp[u], ans); // 每得到一个新子阵，都要更新ans
            }
        }
    }
    cout << ans << '\n';
    return 0;
}