search a 2D matrix

来源:互联网 发布:红警2单位数据 编辑:程序博客网 时间:2024/05/27 06:15

Write an efficient algorithm that searches for a value in an m x n matrix. This matrix has the following properties:

  • Integers in each row are sorted from left to right.
  • The first integer of each row is greater than the last integer of the previous row.

For example,

Consider the following matrix:

[  [1,   3,  5,  7],  [10, 11, 16, 20],  [23, 30, 34, 50]]

Given target = 3, return true.

public class Solution {    public boolean searchMatrix(int[][] matrix, int target) {        if(matrix == null || matrix.length == 0)        {            return false;        }        if(matrix[0] == null || matrix[0].length == 0)        {            return false;        }        int row = matrix.length;        int col = matrix[0].length;        int start = 0;        int end = row - 1;        //find row        while(start + 1 < end)        {            int mid = start + (end-start)/2;            if(matrix[mid][0] == target)            {                return true;            }else if(matrix[mid][0] <= target)            {                start = mid;            }else            {                end = mid;            }        }        if(matrix[end][0] <= target)        {            row = end;        }else        {            row = start;        }                //find column        start = 0;        end = col -1;        while(start + 1 < end)        {            int mid = start + (end - start)/2;            if(matrix[row][mid] == target)            {                return true;            }else if(matrix[row][mid] <= target)            {                start = mid;            }else            {                end = mid;            }        }        if(matrix[row][start] == target)        {            return true;        }else if(matrix[row][end] == target)        {            return true;        }else        {            return false;        }    }}




0 0
原创粉丝点击