FZU2056 最大正方形（二分答案）

Problem 2056 最大正方形

Accept: 171    Submit: 516
Time Limit: 1000 mSec    Memory Limit : 32768 KB

 Problem Description

现在有一个n*m的矩阵A，在A中找一个H*H的正方形，使得其面积最大且该正方形元素的和不大于 limit。

 Input

第一行一个整数T，表示有T组数据。

每组数据 第一行三个非负整数 n m limit

接着 n 行，每行 m 个整数。

0 < n <= 1000, 0 < m <= 1000, 0 <= limit <=100000000 0 <= A[i] <= 1000

 Output

对于每组数据，输出H*H。

 Sample Input

2
2 2 2
1 1
1 1
2 2 4
1 1
1 1

 Sample Output

1
4

 

 

题意： 略

思路： 二分答案 判断一下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int INF = 1e9;
const double eps = 1e-;
const int N = ;
int cas = ;

int n,m;
ll w[N][N],sum[N][N],limit;

void pre()
{
    memset(sum,,sizeof(sum));
    scanf("%d%d%I64d",&n,&m,&limit);
    for(int i=;i<=n;i++)
        for(int j=;j<=m;j++)
        {
            scanf("%I64d",&w[i][j]);
            sum[i][j]=sum[i-][j]+w[i][j];
        }
}

bool judge(int h)
{
    ll s;
    for(int r=;r+h-<=n;r++)
    {
        s=;
        for(int c=;c<=h;c++)
            s+=sum[r+h-][c]-sum[r-][c];
        if(s<=limit) return ;
        for(int c=;c+h<=m;c++)
        {
            s-=sum[r+h-][c]-sum[r-][c];
            s+=sum[r+h-][c+h]-sum[r-][c+h];
            if(s<=limit) return ;
        }
    }
    return ;
}

int solve(int l,int r)
{
    int ans = l;
    while(l<=r)
    {
        int mid = (l+r)/;
        if(judge(mid)) ans=mid,l=mid+;
        else r=mid-;
    }
    return ans*ans;
}

void run()
{
    pre();
    printf("%d\n",solve(,min(n,m)));
}

int main()
{
    #ifdef LOCAL
    freopen("case.txt","r",stdin);
    #endif
    int _;
    scanf("%d",&_);
    while(_--)
        run();
    return ;
}