劍指offer第二版面試題3：二維數組中的查找（JAVA版）

題目：

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

2 4 9 12

4 7 10 13

6 8 11 15

思路：每次取右上角的一個數字（該例子中取9）,若是9大於要找的數字則排除9的那列，不然派出9 的那行

public class Test3 {
    //arr爲要查找數字的數組,num爲要查找的數字
    public static boolean find(int[][] arr,int num){
        boolean flag=false;
        
        if(arr!=null && arr.length!=0){
            int cols=arr.length;//列數
            int rows=arr.length;//行數
            //當前行和當前列(第一次爲最後一個元素)
            int row=0;
            int col=cols-1;
            
            while(row<=rows-1 && col>=0){
                // 比較二維數組中元素與number的關係
                if(arr[row][col]==num){
                    flag=true;
                    break;
                }else if(arr[row][col]>num){
                    col--;
                }else{
                    row++;
                }
            }
        }
        
        return flag;
    }
    public static void main(String[] args) {
        // 測試用例
        int array[][] = { { 1, 2, 8, 9 }, { 2, 4, 9, 12 }, { 4, 7, 10, 13 }, { 6, 8, 11, 15 } };
        boolean flag = find(array, 1);
        System.out.println(flag);
    }
}