28. 搜索二维矩阵

最新推荐文章于 2020-06-10 19:51:13 发布

原创最新推荐文章于 2020-06-10 19:51:13 发布 · 200 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

写出一个高效的算法来搜索 m × n矩阵中的值。

这个矩阵具有以下特性：

每行中的整数从左到右是排序的。

每行的第一个数大于上一行的最后一个整数。

样例

考虑下列矩阵：

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

给出 target = 3，返回 true

public class Solution {
    /*
     * @param matrix: matrix, a list of lists of integers
     * @param target: An integer
     * @return: a boolean, indicate whether matrix contains target
     */
    public static boolean searchMatrix(int[][] matrix, int target) {
        // write your code here
		if (matrix == null) {
			return false;
		}
		if (matrix.length == 0){
		    return false;
		}
		int m = matrix.length-1;
		int n = matrix[0].length-1;
		int row = search1(matrix,target, 0, m);
		return search2(matrix, target, 0, n, row);
		
    }
	
	public static int search1(int[][] matrix, int target, int begin, int end) {
		if (begin <= end) {
			int mid = (begin + end) / 2;
			if (matrix[mid][0] < target) {
				return search1(matrix, target, mid+1, end);
			} else if (matrix[mid][0] > target) {
				return search1(matrix, target, begin, mid-1);
			} else {
				return mid;
			}
		}
		if (begin == 0) {
			return begin;
		}
		return begin-1;
	}
	
	public static boolean search2(int[][] matrix, int target, int begin, int end, int row) {
		if (begin <= end) {
			int mid = (begin + end) / 2;
			if (matrix[row][mid] < target) {
				return search2(matrix, target, mid+1, end, row);
			} else if (matrix[row][mid] > target) {
				return search2(matrix, target, begin, mid-1, row);
			} else {
				return true;
			}
		}
		return false;
	}
}