Leetcode: Smallest Rectangle Enclosing Black Pixels

An image is represented by a binary matrix with 0 as a white pixel and 1 as a black pixel. The black pixels are connected, i.e., there is only one black region. Pixels are connected horizontally and vertically. Given the location (x, y) of one of the black pixels, return the area of the smallest (axis-aligned) rectangle that encloses all black pixels.

For example, given the following image:

[
  "0010",
  "0110",
  "0100"
]
and x = 0, y = 2,
Return 6.

Imagine we project the 2D array to the bottom axis with the rule "if a column has any black pixel it's projection is black otherwise white". The projected 1D array is "0110"

Theorem

If there are only one black pixel region, then in a projected 1D array all the black pixels are connected.

This means we can do a binary search in each half to find the boundaries, if we know one black pixel's position. And we do know that.

To find the left boundary, do the binary search in the [0, y) range and find the first column vector who has any black pixel.

To determine if a column vector has a black pixel is O(m) so the search in total is O(m log n)

We can do the same for the other boundaries. The area is then calculated by the boundaries. Thus the algorithm runs in O(m log n + n log m)

 

follow up是怎么把findLeft和findRight合并成一个：从颜色角度想，其实都是找左边界，只不过分别是找黑色格子左边界和白色格子左边界，所以都是找左边进，一个function就可以，只要控制输入的一个是黑格子一个是白格子

public class Solution {
    public int minArea(char[][] image, int x, int y) {
        int m = image.length;
        int n = image[0].length;
        int left = searchLeft(image, y);
        int right = searchRight(image, y);
        int top = searchTop(image, x);
        int bottom = searchBottom(image, x);
        return (right-left+1)*(bottom-top+1);
    }
    
    public int searchLeft(char[][] image, int right) {
        int l = 0, r = right;
        while (l <= r) {
            int m = (l+r)/2;
            if (!findBlack(image, m, false)) l = m+1;
            else r = m-1;
        }
        return l;
    }
    
    public int searchRight(char[][] image, int left) {
        int l = left, r = image[0].length-1;
        while (l <= r) {
            int m = (l+r)/2;
            if (!findBlack(image, m, false)) r = m-1;
            else l = m+1;
        }
        return r;
    }
    
    public int searchTop(char[][] image, int bottom) {
        int l = 0, r = bottom;
        while (l <= r) {
            int m = (l+r)/2;
            if (!findBlack(image, m, true)) l = m+1;
            else r = m-1;
        }
        return l;
    }
    
    public int searchBottom(char[][] image, int top) {
        int l=top, r=image.length-1;
        while (l <= r) {
            int m = (l+r)/2;
            if (!findBlack(image, m, true)) r = m-1;
            else l = m+1;
        }
        return r;
    }
    
    public boolean findBlack(char[][] image, int cur, boolean row) {
        int m = image.length;
        int n = image[0].length;
        if (row) {
            for (int i=0; i<n; i++) {
                if (image[cur][i] == '1') 
                    return true;
            }
            return false;
        }
        else {
            for (int j=0; j<m; j++) {
                if (image[j][cur] == '1')
                    return true;
            }
            return false;
        }
    }
}

合并findLeft和findRight， 合并findTop和findBottom

public class Solution {
    char[][] image;
    int m;
    int n;
    public int minArea(char[][] iImage, int x, int y) {
        image = iImage;
        m = image.length;
        n = image[0].length;
        int left = searchColumn(0, y, true);
        int right = searchColumn(y, n-1, false);
        int top = searchRow(0, x, true);
        int bottom = searchRow(x, m-1, false);
        return (right-left+1)*(bottom-top+1);
    }
    
    public int searchColumn(int start, int end, boolean isLeft) {
        int l=start, r=end;
        while (l <= r) {
            int mid = (l+r)/2;
            int k;
            for (k=0; k<m; k++) {
                if (image[k][mid] == '1') break;
            }
            if (isLeft) {
                if (k<m) r=mid-1;
                else l=mid+1;
            }
            else {
                if (k<m) l=mid+1;
                else r=mid-1;
            }
        }
        return isLeft? l : r;
    }
    
    public int searchRow(int start, int end, boolean isTop) {
        int l=start, r=end;
        while (l<=r) {
            int mid = (l+r)/2;
            int k;
            for (k=0; k<n; k++) {
                if (image[mid][k] == '1') break;
            }
            if (isTop) {
                if (k<n) r=mid-1;
                else l=mid+1;
            }
            else {
                if (k<n) l=m+1;
                else r=m-1;
            }
        }
        return isTop? l : r;
    }
}