题目描述

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

思路

  1. 判断右上角的数与要查找数字的大小关系
    • 右上角的数大于要查找的数字，由于每一列从上到下是递增的，所有右上角那一列就全部大于要查找的数字，因此去除这一列。
    • 右上角的数小于要查找的数字，那么右上角这一行的数字都小于要查找的数字，则去除这一列。
    • 右上角的数等于要查找的数字，返回 true。
  2. 循环，若没有查找到，返回 false。

代码

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int row = array.size();
        int column = array[0].size();
        int nowRow = 0;
        int nowColumn = column - 1;
        while(nowRow < row && nowColumn >= 0)
        {
            if(target < array[nowRow][nowColumn]) //小于右上角
            {
                nowColumn--;
            }
            else if(target > array[nowRow][nowColumn])
            {
                nowRow++;
            }
            else
            {
                return true;
            }
        }
        return false;
    }
};