LeetCode第167场周赛：5285. 元素和小于等于阈值的正方形的最大边长（前缀和+二分答案）

给你一个大小为 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

思路：首先处理一下前缀和，此前缀和是以点(x,y)为右下角到(0,0)这一块矩形的所有元素的和，然后我们二分边长即可。

class Solution {
	
	private int[][] sum;
	
    public int maxSideLength(int[][] mat, int threshold) {
        
    	int ans=0;
    	int m=mat.length;
    	int n=mat[0].length;
    	sum=new int[m][n];

    	for(int i=0;i<m;i++)
    		for(int j=0;j<n;j++) {
    			if(i==0 && j==0) sum[i][j]=mat[i][j];
    			else if(i==0) sum[i][j]=mat[i][j]+sum[i][j-1];
    			else if(j==0) sum[i][j]=mat[i][j]+sum[i-1][j];
    			else sum[i][j]=mat[i][j]+sum[i-1][j]+sum[i][j-1]-sum[i-1][j-1];
    		}

    	int l=0,r=Math.min(m, n);
    	while(l<=r) {
    		int mid=(l+r)/2;
    		if(jud(mat,m,n,mid,threshold)) {
    			ans=mid;
    			l=mid+1;
    		}
    		else
    			r=mid-1;
    	}
    	
    	return ans;
    }
    
    private boolean jud(int[][] mat,int m,int n,int d,int gate) {
    	if(d==0) return false;
    	int tmp;
    	for(int i=d-1;i<m;i++)
    		for(int j=d-1;j<n;j++) {
    			if(i==d-1 && j==d-1) tmp=sum[i][j];
    			else if(i==d-1) tmp=sum[i][j]-sum[i][j-d];
    			else if(j==d-1) tmp=sum[i][j]-sum[i-d][j];
    			else
    				tmp=sum[i][j]-sum[i-d][j]-sum[i][j-d]+sum[i-d][j-d];
    			if(tmp<=gate)
    				return true;
    		}
    	return false;
    }
}