LeetCode 221. Maximal Square
题意:
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 0
思路: 这道题被我想成了单调栈了~ 统计每个位置,从上往下连续的1的数目。 然后对于每一行找出最大的正方形。 利用单调栈。 单调栈弹出的时候要把当前的值更新到被弹出的位置。
犯的错误: 1、没意识到数据是char,一直当成了int
2、单调栈弹出的时候忘记更新~
3、手搓, >= 写成>
另解: 想了想这道题难度是medium。 按道理不应该用单调栈。 差了下, 可以用dp, 对于1的位置,dp[i][j] = min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + 1
老矣,没想出来。
AC代码:
class Solution {
public:
int maximalSquare(vector < vector<char> >& matrix) {
int n = matrix.size();
if(n == 0) return 0;
int m = matrix[0].size();
if(m == 0) return 0;
vector<int> temp(m+1, 0);
vector< vector<int> >data(n, temp);
for(int i=0; i<m; i++)
data[0][i] = matrix[0][i]-'0';
for(int i=1; i<n; i++)
{
for(int j=0; j<m; j++)
{
if(matrix[i][j] == '0')
data[i][j] = 0;
else data[i][j] = data[i-1][j] + 1;
}
}
int ans = 0;
for(int i=0; i<n; i++)
{
data[i][m] = 0;
stack<int>s;
if(data[i][0] >= 1)
ans = max(ans, 1);
s.push(0);
for(int j=1; j<=m; j++)
{
if (data[i][s.top()]<data[i][j])
{
s.push(j);
continue;
}
int last;
while(s.size()>0 && data[i][s.top()]>=data[i][j])
{
last = s.top();
s.pop();
ans = max(ans, min(data[i][last], j-last) );
}
if(s.size() == 0)
ans = max(ans, min(data[i][j], j+1) );
data[i][last] = data[i][j];
s.push(last);
}
}
return ans*ans;
}
};

浙公网安备 33010602011771号