一直没想到合适的处理方式,又看了题解。(不算原创了-_-#)
以dp[i][j]为存储左上角矩形的全部元素和:
dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+a[i][j]
然后从大到小枚举正方形边长,利用s=dp[i][j]-d[i-h][j]-dp[i][j-h]+dp[i-h][j-h](画个图很容易得到)
计算以(i,j)为右下角顶点,边长为h的正方形的全部元素和。
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxn =1000+5;
int dp[maxn][maxn];
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
int n,m,limit,a;
scanf("%d%d%d",&n,&m,&limit);
memset(dp,0,sizeof(dp));
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++){
scanf("%d",&a);
dp[i][j]=dp[i-1][j]+dp[i][j-1]-dp[i-1][j-1]+a;
}
int H=min(n,m),h;
for(h=H;h>=1;h--){
int ans=-1;
for(int i=n;i>=h;i--)
for(int j=m;j>=h;j--){
int t=dp[i][j]-dp[i-h][j]-dp[i][j-h]+dp[i-h][j-h];
if(t<=limit&&t>ans) ans=t;
}
if(ans!=-1) break;
}
printf("%d\n",h*h);
}
return 0;
}