劍指Offer-二維數組中的查找

題目描述

在一個二維數組中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序排序。請完成一個函數，輸入這樣的一個二維數組和一個整數，判斷數組中是否含有該整數。

思路

思路一：

暴力，時間復雜度$O(mn)$

思路二：

利用二維數組由上到下，由左到右遞增的規律，那么選取左下角或者右上角的元素a[i][j]與target進行比較，當target大於元素a[i][j]時，那么target必定在元素a所在行的右邊,即j++；當target大於元素a[i][j]時，那么target必定在元素a所在列的上邊,即i--；時間復雜度$O(m+n)$

思路三：

每一行都按照從左到右遞增的順序排序，把每一行看作有序遞增數組，利用二分查找，通過遍歷每一行查找得到答案，時間復雜度$mlog(n)$

代碼實現

package Array;

/**
 * 二維數組中的查找
 * 在一個二維數組中，每一行都按照從左到右遞增的順序排序，每一列都按照從上到下遞增的順序排序。
 * 請完成一個函數，輸入這樣的一個二維數組和一個整數，判斷數組中是否含有該整數。
 */
public class Solution11 {
    public static void main(String[] args) {
        Solution11 solution11 = new Solution11();
        int[][] arr = new int[][]{{1, 2, 3, 4, 5}, {2, 4, 7, 8, 10}};
        System.out.println(solution11.Find_2(7, arr));
    }

    /**
     * 每一行都按照從左到右遞增的順序排序，把每一行看作有序遞增數組
     * 利用二分查找
     * 通過遍歷每一行查找得到答案
     * 時間復雜度mlog(n)
     *
     * @param target
     * @param array
     * @return
     */
    public boolean Find_3(int target, int[][] array) {
        if (array == null || array.length == 0 || (array.length == 1 && array[0].length == 0)) return false;
        for (int i = 0; i < array.length; i++) {
            int begin = 0;
            int end = array[0].length - 1;
            while (begin <= end) {
                int mid = (begin + end) / 2;
                if (target > array[i][mid]) {
                    begin = mid + 1;
                } else if (target < array[i][mid]) {
                    end = mid - 1;
                } else {
                    return true;
                }
            }

        }
        return false;
    }

    /**
     * 利用二維數組由上到下，由左到右遞增的規律，
     * 那么選取左下角或者右上角的元素a[i][j]與target進行比較，
     * 當target大於元素a[i][j]時，那么target必定在元素a所在行的右邊,
     * 即j++；
     * 當target大於元素a[i][j]時，那么target必定在元素a所在列的上邊,
     * 即i--；
     * 時間復雜度m+n
     *
     * @param target
     * @param array
     * @return
     */
    public boolean Find_2(int target, int[][] array) {
        if (array == null || array.length == 0 || (array.length == 1 && array[0].length == 0)) return false;
        int i = array.length - 1;
        int j = 0;
        while (i >= 0 && j <= array[0].length) {
            if (target > array[i][j]) {
                j++;
            } else if (target < array[i][j]) {
                i--;
            } else {
                return true;
            }
        }
        return false;
    }

    /**
     * 暴力
     * 時間復雜度mn
     *
     * @param target
     * @param array
     * @return
     */
    public boolean Find(int target, int[][] array) {
        if (array == null || array.length == 0 || (array.length == 1 && array[0].length == 0)) return false;
        for (int i = 0; i < array.length; i++) {
            for (int j = 0; j < array[0].length; j++) {
                if (target == array[i][j]) {
                    return true;
                }
            }
        }
        return false;
    }
}