剑指offer面试题4 二维数组中的查找

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

思路：先横向查找，再竖向查找，根据矩阵的规则，快速排除不可能的矩阵块，时间复杂度o(col+row):

改进版本可以加入二分查找的思想，达到log(col)+log(row)的复杂度

#include <iostream>
#include <string>
#include <memory>
#include <vector>
#include<cmath>
#include<algorithm>
#include <iomanip>
using namespace std;

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        if(array.size()==0) {
            return false;
        }
        int row=0;
        int col=0;
        int rowBound=array.size();
        int colBound=array[0].size();
        while(row<rowBound&&row>=0&&col<colBound&&col>=0) {
            if(array[row][col]==target) {
                return true;
            } else if(array[row][col]<target) {
                if(col+1<colBound&&array[row][col+1]<=target) {
                    col++;
                } else {
                    if(row+1<rowBound&&array[row+1][col]<=target) {
                        row++;
                    } else {
                        colBound=col;
                        col--;
                    }  
                }
            } else {
                return false;
            }
        }
        return false;
    }
};

int main()
{
    vector<int> test1={1,2,3,4};
    vector<int> test2={7,8,9,10};
    vector<int> test3={12,13,14,15};
    vector<vector<int>> test;
    test.push_back(test1);
    test.push_back(test2);
    test.push_back(test3);
    Solution sol;
    int target;
    while (true)
    {
        cin>>target;
        bool res=sol.Find(target,test);
        cout<<res<<endl;
    }
    system("pause");
    return 0;
}