原题地址:https://leetcode-cn.com/problems/maximal-square/submissions/
题目描述:
在一个由 0 和 1 组成的二维矩阵内,找到只包含 1 的最大正方形,并返回其面积。
示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4
解题方案:
本题是动态规划的题型,找到通项公式就可以了。寻找动态规划的规律是解题的关键,这种规律比较灵活,需要见多识广才能很快的找出答案。
若当前[i, j]节点的值为0时,dp值为0;若当前[i, j]节点的值为1时,需要进行判断[i-1, j]节点、[i, j -1]节点和[i-1, j-1]节点的值,去其中dp的最小值,当前节点的dp值为所取最小值加一。
代码:
class Solution {
public:
int maximalSquare(vector<vector<char>>& matrix) {
if(matrix.empty() || matrix[0].empty())
return 0;
int m = matrix.size();
int n = matrix[0].size();
int res = 0;
vector<vector<int>> dp(m, vector<int>(n, 0));
for(int i = 0; i < m; ++i) {
for(int j = 0; j < n; ++j) {
if(i == 0 || j == 0)
dp[i][j] = matrix[i][j] - '0';
else if(matrix[i][j] == '1')
dp[i][j] = min(dp[i-1][j], min(dp[i][j-1], dp[i-1][j-1])) + 1;
res = max(res, dp[i][j]);
}
}
return res * res;
}
};