1292 元素和小于等于阈值的正方形的最大边长

题目描述:
给你一个大小为 m x n 的矩阵 mat 和一个整数阈值 threshold。
请你返回元素总和小于或等于阈值的正方形区域的最大边长;如果没有这样的正方形区域,则返回 0 。

示例 1:
输入:mat = [[1,1,3,2,4,3,2],[1,1,3,2,4,3,2],[1,1,3,2,4,3,2]], threshold = 4
输出:2
解释:总和小于或等于 4 的正方形的最大边长为 2,如图所示。

示例 2:
输入:mat = [[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2],[2,2,2,2,2]], threshold = 1
输出:0

示例 3:
输入:mat = [[1,1,1,1],[1,0,0,0],[1,0,0,0],[1,0,0,0]], threshold = 6
输出:3

示例 4:
输入:mat = [[18,70],[61,1],[25,85],[14,40],[11,96],[97,96],[63,45]], threshold = 40184
输出:2

提示:
1 <= m, n <= 300
m == mat.length
n == mat[i].length
0 <= mat[i][j] <= 10000
0 <= threshold <= 10^5

方法1:
主要思路:解题链接汇总
(1)前缀和+二分;
(2)先生成二维的前缀和;
(3)再对正方形的宽度进行二分,搜索满足要求的结果;

class Solution {
public:
    bool is_valid(vector<vector<int>>&sub_sum,int len,int m,int n,int threshold){//判断是否存在满足要求的给定长度的正方形
        for(int i=len;i<=m;i++){
            for(int j=len;j<=n;j++){
                if(sub_sum[i][j]-sub_sum[i-len][j]-sub_sum[i][j-len]+sub_sum[i-len][j-len]<=threshold){
                    return true;
                }
            }
        }
        return false;
    }
    int maxSideLength(vector<vector<int>>& mat, int threshold) {
        int m=mat.size();
        int n=mat[0].size();
        vector<vector<int>> sub_sum(m+1,vector<int>(n+1,0));//前缀和
        for(int i=1;i<=m;i++){
            for(int j=1;j<=n;j++){
                sub_sum[i][j]=sub_sum[i-1][j]+sub_sum[i][j-1]+mat[i-1][j-1]-sub_sum[i-1][j-1];
            }
        }
        //二分
        int left=1,right=min(m,n);
        int res=0;
        while(left<=right){
            int mid=left+(right-left)/2;
            if(is_valid(sub_sum,mid,m,n,threshold)){
                res=mid;
                left=mid+1;
            }else{
                right=mid-1;
            }
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值