最大子阵
题目描述
给定一个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;
}