1461-矩阵问题(前缀和+二分答案)

题目描述
给定一个n*n的矩阵，你可以从矩阵中任选一块正方形，但必须保证该正方形中所有数字之和小于等于m。
请问以(i,j)为左上角且符合以上要求的正方形中，边长最大的是多少？
输入
第一行两个正整数n,m，其中n<=1000,m<=10^9。
接下来n行，每行n个非负整数aij，aij<=10^9。
输出
输出n行，每行n个数字。其中第i行第j个数字表示，以(i,j)为左上角且符合题目要求的正方形的最大边长。若以(i,j)为左上角没有符合要求的正方形，则输出0。
样例输入
3 4
1 1 1
1 1 1
1 1 1
样例输出
2 2 1
2 2 1
1 1 1

#include <bits/stdc++.h>
using namespace std;
const int N = 1010;
int A[N][N];
long long pre[N][N];			///元素单个的范围达到1e9，故开一个long long
int n,m;
long long _count(int x1,int y1,int x2,int y2){
    long long sum;
    sum = pre[x2][y2]-pre[x1-1][y2]-pre[x2][y1-1]+pre[x1-1][y1-1];	///求出以x1、y1位左上角，x2、y2为右下角的前缀和。
    return sum;
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            scanf("%d",&A[i][j]);
            pre[i][j] = pre[i-1][j]+pre[i][j-1]-pre[i-1][j-1]+A[i][j];   ///初始化矩阵前缀和
        }
    }
    for(int i=1;i<=n;i++){                                              ///枚举左上角坐标
        for(int j=1;j<=n;j++){
            int l=0,r=min(n-i+1,n-j+1),mid;                             ///二分正方形边长
            while(l<r){
                mid = l+r+1>>1;
                if(_count(i,j,i+mid-1,j+mid-1)<=m)l=mid;
                else r=mid-1;
            }
            printf("%d ",l);
        }
        puts("");
    }
    return 0;
}