剑指04 ---- 二维数组中的查找

class Solution {

public:

    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) 

    {

        if(matrix.empty())

            return false;

        int m = matrix[0].size();

        for(int i = 0;i < matrix.size();i++)

        {

            for(int j = 0;j < m;j++)

            {

                if(target == matrix[i][j])

                    return true;

                if(target < matrix[i][j])

                {

                    if(j == 0)

                        return false;

                    else

                    {

                        m = j;

                        break;

                    }

                }

            }

        }

        return false;

    }

};

算法思路:

如果数组为空，则直接返回false

若数组不空，利用两重循环，如果找到了元素，直接返回true；

若目标值小于第1列的元素可以直接返回false；

若目标小于第i行第j列的元素，则说明接下来遍历到的这一列的元素都会比目标值大，所以将第二层循环的遍历终点进行更改，跳出循环。