给你一个 m x n 的矩阵 matrix 和一个整数 k ,找出并返回矩阵内部矩形区域的不超过 k 的最大数值和。
题目数据保证总会存在一个数值和不超过 k 的矩形区域。
示例 1:
输入:matrix = [[1,0,1],[0,-2,3]], k = 2
输出:2
解释:蓝色边框圈出来的矩形区域 [[0, 1], [-2, 3]] 的数值和是 2,且 2 是不超过 k 的最大数字(k = 2)。
方法:
使用了前缀和,首先求出((0,0),(x,y))这个区域内的数值和,之后利用此结果,求所有可能矩形的数值和。使用暴力的方法,数据太弱,并没有超时。
思路不难,但是有一些小细节需要注意:
(1)矩形不能为空
(2)在边界时的处理
代码:
const int maxn =200;
int sum[maxn][maxn];
class Solution {
public:
int maxSumSubmatrix(vector<vector<int>>& matrix, int k) {
memset(sum,0,sizeof(sum));
for(int i=0;i<matrix.size();i++)
{
for(int j = 0; j < matrix[i].size(); j++)
{
if(i==0){
if(j==0){
sum[i][j]=matrix[i][j];
}
else{
sum[i][j]=sum[i][j-1]+matrix[i][j];
}
}
else{
if(j==0){
sum[i][j]=sum[i-1][j]+matrix[i][j];
}
else{
sum[i][j]=sum[i][j-1]+sum[i-1][j]-sum[i-1][j-1]+matrix[i][j];
}
}
}
}
int ans=-1000000000;
for(int i =0 ;i<matrix.size(); i++)
{
for( int j = 0; j<matrix[i].size() ;j++){
int tmp=sum[i][j];
if(tmp>ans && tmp<=k){
ans=tmp;
}
for (int ii = i;ii<matrix.size();ii++){
for(int jj = j;jj<matrix[ii].size();jj++){
if((ii==i) &&(jj==j)){
continue;
}
if((ii!=i) &&(jj !=j)){
tmp=sum[ii][jj]-sum[ii][j]-sum[i][jj]+sum[i][j];
}
else if(ii ==i)
{
tmp=sum[ii][jj]-sum[i][j];
}
else if(jj=j){
tmp=sum[ii][jj]-sum[i][j];
}
if(tmp>ans && tmp<=k){
ans=tmp;
}
}
}
}
}
return ans;
}
};