在一个由 0 和 1 组成的二维矩阵内,找到只包含 1 的最大正方形,并返回其面积。
示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4
思路
问题要找到最大全为1的正方形。p(i,j)点为右下角的正方形,计算全为1的正方形的边长。很明显与p(i-1, j), p(i-1, j-1), p(i, j-1)三个点有关。
p(i,j) = 三个点中的最小值加1.。画个图更好理解。
int maximalSquare(vector<vector<char>>& matrix) {
//dp[i][j] 以i,j为右下角的正方形的边长
int rows = matrix.size();
if(rows==0)
{
return 0;
}
int cols = matrix[0].size();
vector<vector<int>> dp(rows, vector<int>(cols));
int maxsize = 0;
for(int i = 0; i < rows; ++i)
{
for(int j = 0; j < cols; ++j)
{
if(matrix[i][j] == '1')
{
int left = max(0, i-1);
int top = max(0, j-1);
dp[i][j] = min(dp[left][j], min(dp[i][top], dp[left][top])) + 1;
}
maxsize = max(maxsize, dp[i][j]);
}
}
return maxsize*maxsize;
}