蓝桥杯基础算法训练——最大子阵

最大子阵

问题描述

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

输入格式

输入的第一行包含两个整数 $n,m$ 分别表示矩阵A的行数和列数。$（1\le n,m\le 50）$
接下来 $n$ 行，每行 $m$ 个整数，表示矩阵$A_{i,j}(-1000\le A_{i,j}\le 1000)$

输出格式

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

解题思路

#include<iostream>
using namespace std;
int a[51][51], b[51][51];
int main() 
{
	int n, m;
	int max = -10000;
	cin >> n >> m;
	for (int i = 1; i <= n; i++) 
	{
		for (int j = 1; j <= m; j++) 
		{
			cin >> a[i][j];
			b[i][j] = b[i - 1][j] + a[i][j];
		}
	}
	for (int i = 1; i <= n; i++) 
	{
		for (int j = i; j <= n; j++) 
		{
			int sum = 0;
			for (int k = 1; k <= m; k++) 
			{
				sum += b[j][k] - b[i - 1][k];
				if (sum > max) max = sum;
				if (sum < 0) sum = 0;
			}
		}
	}
	cout << max << endl;
	return 0;
}