[剑指offer]面试题3（二维数组中的查找）

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

基本思路：
首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数字，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。
比较以下两个版本：

for循环：

#include <stdio.h>

bool find(int a[][4], int n)
{
    int row;    //行数
    int col;    //列数

    for (row = 0; row <= 3;)
    {
        for (col = 3; col >= 0;)
        {
            if (n == a[row][col])
            {
                printf("row = %d, col = %d\n", row + 1, col + 1);
                return true;
            }
            if (n < a[row][col])
                col--;          
            if (n > a[row][col])
                row++;
        }
    }
    //if ((row > 3) || (col < 0))
        return false;
}

int main()
{
    int a[4][4] = {{1,3,8,9},{2,4,9,12},{4,7,10,13},{6,8,11,15}};

    printf("请输入要查找的数:\n");

    int n;
    scanf("%d", &n);

    if (find(a, n) == true)
        printf("true\n");
    else
        printf("false\n");

    return 0;
}

while循环：

#include <stdio.h>

bool find(int a[][4], int n)
{
    int row = 0;    //行数
    int col = 3;    //列数
    bool flag = false;

    while(row < 4 && col >= 0)
    {
        if (a[row][col] == n)
        {
            flag = true;
            printf("row = %d, col = %d\n", row + 1, col + 1);
            return flag;
        }
        else if (a[row][col] > n)
        {
            col--;
        }
        else
        {
            row++;
        }
    }
}

int main()
{
    int a[4][4] = {{1,3,8,9},{2,4,9,12},{4,7,10,13},{6,8,11,15}};

    printf("请输入要查找的数:\n");

    int n;
    scanf("%d", &n);

    if (find(a, n) == true)
        printf("Yes\n");
    else
        printf("No\n");

    return 0;
}

在for循环程序测试中发现如果输入的数比数组中最小的数还小， 那么结果不是No，而是程序跳不出循环一直卡着，而使用while循环则没有这样的问题。
那么，要怎么改进使for循环能正常呢？

#include <stdio.h>

bool find(int a[][4], int n)
{
    int row;    //行数
    int col;    //列数
    bool flag = false;

    for (row = 0; row <= 3;)
    {
        for (col = 3; col >= 0;)
        {
            if (n == a[row][col])
            {
                printf("row = %d, col = %d\n", row + 1, col + 1);
                flag = true;
                return flag;
            }
            if (n < a[row][col])
                col--;          
            if (n > a[row][col])
                row++;
        }
        if (col < 0)   //注意这里，如果一直没有找到比自己小的数，要跳出循环
        {
            break;
        }       
    }
    return false;
}

int main()
{
    int a[4][4] = {{1,3,8,9},{2,4,9,12},{4,7,10,13},{6,8,11,15}};

    printf("请输入要查找的数:\n");

    int n;
    scanf("%d", &n);

    if (find(a, n) == true)
        printf("Yes\n");
    else
        printf("No\n");

    return 0;
}