动态规划,dp[i][j]代表以i,j为右下角的正方形的最长边长。代码如下:
int maximalSquare(vector<vector<char>>& matrix) {
int result = 0;
if(matrix.empty())
return result;
const int n = matrix.size();
const int m = matrix[0].size();
vector<vector<int> > dp(n + 1, vector<int>(m + 1, 0));
for(int i = 0; i < n; ++i)
for(int j = 0; j < m; ++j){
if(matrix[i][j] == '1'){
dp[i + 1][j + 1] = min(dp[i][j], min(dp[i + 1][j], dp[i][j + 1])) + 1;
result = max(result, dp[i + 1][j + 1]);
}
}
return result * result;
}
利用回滚数组可将空间优化到一维!
若输入matrix类型为int,dp数组可重用matrix,空间复杂度可优化到常数空间。