题目:
Given a 2D binary matrix filled with 0's and 1's, find the largest square containing all 1's and return its area.
For example, given the following matrix:
1 0 1 0 0 1 0 1 1 1 1 1 1 1 1 1 0 0 1 0Return 4.
分析:动态规划,从左上角开始,如果当前位置为1,那么到当前位置包含的最大正方形边长为左/左上/上的值中的最小值加一,因为边长是由短板控制的。注意返回的是面积,不要因为小问题而出错。
代码:
class Solution {
public:
int maximalSquare(vector<vector<char>>& matrix) {
int ret=0;
if(matrix.empty()||matrix[0].empty()){
return ret;
}
int m=matrix.size();
int n=matrix[0].size();
vector<vector<int> > v(m,vector<int>(n,0));
for(int i=0;i<m;i++){
if(matrix[i][0]=='1'){
v[i][0]=1;
ret=1;
}
}
for(int j=0;j<n;j++){
if(matrix[0][j]=='1'){
v[0][j]=1;
ret=1;
}
}
for(int i=1;i<m;i++){
for(int j=1;j<n;j++){
if(matrix[i][j]=='1'){
v[i][j]=min(v[i-1][j],min(v[i-1][j-1],v[i][j-1]))+1;
ret=max(ret,v[i][j]);
}
}
}
return ret*ret;
}
};