java杨氏矩阵查找算法

 

在一个m行n列二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

定位法，时间复杂度O（m+n）。首先直接定位到最右上角的元素，再配以二分查找，比要找的数（6）大就往左走，比要找数（6）的小就往下走，直到找到要找的数字（6）为止，如下图所示

 

 

代码如下

package contcurrentandalgorithm;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

/**
 *
 * @author Administrator
 * zyyjiao@mail.ustc.edu.cn
 */
public class YangShiJuZheng {

    public static void main(String[] args) throws IOException {

        int a[][] = {{1, 2, 8, 9}, {2, 4, 9, 12}, {4, 7, 10, 13}, {6, 8, 11, 15}};
        for (int search = 1; search <= 15; search++) {
            BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
            String re = br.readLine();
            search = Integer.parseInt(re);
            if (yang(a, search)) {
                System.out.println("存在");
            } else {
                System.out.println("不存在");
            }
        }
    }

    public static boolean yang(int a[][], int search) {
        int i = 0;
        int j = 3;
        int ROW = 4;
        int COL = 4;
        int var = a[i][COL - 1];
        while (true) {
            if (var == search) {
                return true;
            } else if (var < search && i < ROW - 1) {
                var = a[++i][j];
            } else if (var > search && j > 0) {
                var = a[i][--j];
            } else {
                return false;
            }
        }
    }
}