题目描述
在一个由 0 和 1 组成的二维矩阵内,找到只包含 1 的最大正方形,并返回其面积。
示例:
输入:
1 0 1 0 0
1 0 1 1 1
1 1 1 1 1
1 0 0 1 0
输出: 4
分析
本题考虑使用动态规划。定义状态dp[i][j]是以matrix[i][j]为右下角的最大的正方形的边长。那么状态转移方程就是:
d
p
[
i
]
[
j
]
=
m
i
n
(
d
p
[
i
−
1
]
[
j
]
,
d
p
[
i
]
[
j
−
1
]
,
d
p
[
i
−
1
]
[
j
−
1
]
)
+
1
dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
dp[i][j]=min(dp[i−1][j],dp[i][j−1],dp[i−1][j−1])+1
代码
class Solution {
public:
int maximalSquare(vector<vector<char>>& matrix) {
int m = matrix.size();
if(m==0) return 0;
int n = matrix[0].size();
int res = 0;
//使用动态规划
vector<vector<int> > dp(m, vector<int>(n, 0));
for(int i=0;i<n;i++){
if(matrix[0][i]=='1') dp[0][i] = 1;
res = max(res, dp[0][i]);
}
for(int i=0;i<m;i++){
if(matrix[i][0]=='1') dp[i][0] = 1;
res = max(res, dp[i][0]);
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
if(matrix[i][j]=='0') dp[i][j] = 0;
else dp[i][j] = min(min(dp[i-1][j], dp[i][j-1]), dp[i-1][j-1]) + 1;
res = max(res, dp[i][j]);
}
}
return res * res;
}
};