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;
    }
};

 

posted @ 2016-03-19 19:54  Gu Feiyang  阅读(117)  评论(0)    收藏  举报