矩形区域不超过 K 的最大数值和
1.问题描述
给定一个非空二维矩阵 matrix 和一个整数 k,找到这个矩阵内部不大于 k 的最大矩形和。
示例:
输入: matrix = [[1,0,1],[0,-2,3]], k = 2
输出: 2
解释: 矩形区域 [[0, 1], [-2, 3]] 的数值和是 2,且 2 是不超过 k 的最大数字(k = 2)。
说明:
矩阵内的矩形区域面积必须大于 0。
如果行数远大于列数,你将如何解答呢?
[OJ中有个样例就是1300行 13列 ,注意超时]
2.输入说明
首先输入matrix的行数m、列数n,
然后输入m行,每行n个整数。
最后输入一个整数k。
3.输出说明
输出一个整数。
4.范例
输入
2 3
1 0 1
0 -2 3
2
输出
2
5.代码
#include <iostream>
#include <queue>
#include <cstdlib>
#include <cstring>
#include<algorithm>
#include<set>
using namespace std;
int findMaxSum(vector<vector<int> >matrix, int k)
{
//思想参考:https://leetcode.cn/problems/max-sum-of-rectangle-no-larger-than-k/solution/gong-shui-san-xie-you-hua-mei-ju-de-ji-b-dh8s/
if (matrix.empty() || matrix[0].empty()) return 0;//特殊情形判断
int m = matrix.size();//行
int n = matrix[0].size();//列
int res = INT_MIN;
int sum[m][n];//这里使用vector部分样例会超时
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
int t = matrix[i][j];
if (i > 0) t += sum[i - 1][j];
if (j > 0) t += sum[i][j - 1];
if (i > 0 && j > 0) t -= sum[i - 1][j - 1];
sum[i][j] = t;
for (int r = 0; r <= i; ++r) {
for (int c = 0; c <= j; ++c) {
int d = sum[i][j];
if (r > 0) d -= sum[r - 1][j];
if (c > 0) d -= sum[i][c - 1];
if (r > 0 && c > 0) d += sum[r - 1][c - 1];
if (d <= k) res = max(res, d);
}
}
}
}
return res;
}
int main()
{
vector<vector<int> > nums;
int m,n,k,tmp;
cin >> m >>n;
for (int i = 0; i < m; i++)
{
vector<int> t;
t.clear();
for (int j = 0; j < n; j++)
{
cin >> tmp;
t.push_back(tmp);
}
nums.push_back(t);
}
cin >> k;
int res = findMaxSum(nums, k);
cout << res << endl;
return 0;
}