C 二維矩陣查找數字

1.題目

//楊氏矩陣
有一個二維數組.
數組的每行從左到右是遞增的，每列從上到下是遞增的.
在這樣的數組中查找一個數字是否存在。
時間複雜度小於O(N);

2.程序代碼

#define _CRT_SECURE_NO_WARNINGS 1

#include <stdio.h>
#include <windows.h>

int find_num(const int arr[4][4], int num, int row, int col)
{
    int i = 0;
    int j = 0;

    while ((num > arr[0][0]) && (num < arr[row - 1][col - 1])&&
    (i < row) && (j < col))//輸入的數字要滿足在數組最大數字和最小數字中間
                    //i和j要不超過數組的範圍
    {
        if (num > arr[i][col - 1 - j])
        {
            i++;//如果要找的數比這行最後一個元素大，則跳到下一行
        }
        else if (num < arr[i][col - 1 - j])
        {
            j++;//如果要找的數比這行最後一個元素小，則跳到前一個元素
        }
        else if (num == arr[i][col - 1 - j])
        {
            return 1;//如果找到了，就返回一
        }
    }

    return 0;//當要找的數字不在最小值到最大值之間，或者當i，j，超出範圍則返回0
}

int main()
{
    int arr[4][4] = { 1, 3, 5, 7, 9, 11, 13, 15, 17, 
                 19, 21, 23, 25, 27, 29, 31 };//定義一個二維數組

    int num = 0;//定義參數存放輸入的值

    printf("please input a number:");
    scanf("%d", &num);

    if (find_num(arr, num, 4, 4))//如果自定義函數返回值爲1，則找到，相反則沒找到
    {
        printf("find\n");
    }
    else
    {
        printf("not find\n");
    }

    system("pause");
    return 0;
}

3.執行結果