题目描述:
在给定的二维01数组中,找到1能构成的最大面积的正方形,返回面积。
思路:
动态规划。以 i, j 为右下方点的正方形边长,取决于其上、左、右三个点的正方形长度。
代码:
class Solution {
public:
int maximalSquare(vector<vector<char>>& matrix) {
if (matrix.empty()) return 0;
int n = matrix.size();
int m = matrix[0].size();
vector<vector<int> > dp(n, vector<int>(m, 0));
int ans = 0;
for (int i=0; i<n; ++i) {
dp[i][0] = (matrix[i][0] == '1' ? 1 : 0);
ans = max(ans, dp[i][0]);
}
for (int i=0; i<m; ++i) {
dp[0][i] = (matrix[0][i] == '1' ? 1 : 0);
ans = max(ans, dp[0][i]);
}
for (int i=1; i<n; ++i) {
for (int j=1; j<m; ++j) {
if (matrix[i][j] == '1') {
dp[i][j] = min(dp[i-1][j-1], min(dp[i-1][j], dp[i][j-1])) + 1;
ans = max(ans, dp[i][j]);
}
}
}
return ans * ans;
}
};
读错题。写了很久的dfs。
一个bug改一下午,终于跑起来实验了,早知道周末不看剧了!!嘤。