一个二维数组中的值只有0和1,现在想要找出这个数组中的最大的1所构成的正方形, 要求输出这个正方形的面积
这是华为笔试的一道题目, 拿到这个题的第一个想法就是循环套循环, 但是发现时间复杂度太高了,无法AC, 后面也没有做出来这一题, 笔试结束后在网上看了下, 大家给出的是动态规划的方法来做, 现在将这个算法整理如下;
首先, 定义一个和原来的矩阵matrix同等大小的矩阵dp[maxn][maxn], dp[i][j]表示matrix中以i,j为右下角的矩阵的边长,
初始时候,将matrix的第0行和第0列都赋值到dp中,然后,状态转移方程如下:
若matrix[i][j]=0, 则dp[i][j]=0;
若matrix[i][j]=1,则dp[i][j]= min(dp[i-1][j],dp[i-1][j-1],dp[i][j-1]) +1;
全部代码如下:
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 10;
int getsquare(int matrix[][10], int n) {
int dp[maxn][maxn];
for (int i = 0; i < n; i++) {
dp[0][i] = matrix[0][i];
dp[i][0] = matrix[i][0];
}
for (int i = 1; i < 4; i++) {
for (int j = 1; j < 4; j++) {
if (matrix[i][j] == 0) dp[i][j] = 0;
else {
dp[i][j] = min(dp[i - 1][j], min(dp[i][j - 1], dp[i - 1][j - 1])) + 1;
}
}
}
int m = 0;
for (int i = 0; i < 4; i++) {
for (int j = 0; j < 4; j++) {
if (m < dp[i][j]) m = dp[i][j];
}
}
return m * m;
}
int main() {
int matrix[maxn][maxn] = { 0 };
int dp[maxn][maxn] = { 0 };
int n = 0;
cin >> n;
for (int i = 0; i < n; i++) {
for (int j = 0; j < n; j++) {
cin >> matrix[i][j];
}
}
cout<< getsquare(matrix, n);
}