题目链接:
221. 最大正方形 - 力扣(LeetCode)leetcode-cn.com题目描述:
在一个由 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 的最大正方形,我们可以用动态规划解决。用 dp[i][j] 表示直到 (i, j) 位置的可以包含最大正方形的边长,注意这里的正方形需要包含 (i, j) 位置。
在矩形左侧和上边界,如果 matrix[i][j]=='1',有 dp[0][j]==dp[i][0]==1,即能构成的最大正方形的边长为 1。
而对于更一般的情况:
- 如果 matrix[i][j]=='0',显然 dp[i][j]==0。
- 如果 matrix[i][j]=='1',状态转移方程为:dp[i][j]=min(min(dp[i-1][j], dp[i][j-1]), dp[i-1][j-1])+1,即 (i, j) 位置的值被其左侧、上侧、左上侧的值所限制。
代码如下:
class Solution {
public:
int maximalSquare(vector<vector<char>>& matrix) {
int row = matrix.size();
if(row < 1) return 0;
int col = matrix[0].size();
if(col < 1) return 0;
vector<vector<int>> dp(row, vector<int>(col, 0));
int res = 0;
for (int i = 0; i < row; ++i) {
for (int j = 0; j < col; ++j) {
if (i==0 || j==0) dp[i][j] = matrix[i][j]-'0';
else if(matrix[i][j]=='1') {
dp[i][j] = min(min(dp[i-1][j], dp[i][j-1]), dp[i-1][j-1]) + 1;
}
else {
dp[i][j] = 0;
}
res = max(res, dp[i][j]);
}
}
return res * res;
}
};
如果有任何疑问,欢迎提出。如果有更好的解法,也欢迎告知。