【剑指Offer】——04二维数组中的查找

题目描述：二维数组中的查找

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

例如：数组

1	2	8	9
2	4	9	12
4	7	10	13
6	8	11	15

 

查找数字7，则为真，查找数字5，则为假

解题要点：观察数组的特点，右上角的数字：每列的最小值，每行的最大值。由此可以缩小查找范围

解析：设置查找的索引下标，初始值：数组的右上角

           查找值 ＞右上角值：行索引下移  （比该行的最大值都大，舍弃该行）

           查找值＜右上角值：列索引左移（比该列的最小值都小，舍弃该列）

           若右上角的元素值＝查找值  则返回真

            若索引下标超限，则 返回假

代码:

#include<iostream>
#include<vector>
using namespace std;

bool findNum(vector<vector<int> > a, int num){
    int i = 0, j = a[0].size() - 1 ;

    while(i < a.size() && j >= 0){
        if(a[i][j] == num)
            return true;
        if(a[i][j] < num)
            i++;
        if(a[i][j] > num)
            j--;
    }

    return false;
}

int main(){
    vector<vector <int> > a;
    vector<int> aa;
    int temp,m,n,num;
    bool res;

    cout << "m = " ;
    cin >> m ;
    cout << "n = " ;
    cin >> n ;
    cout << "a :" << endl;

    for(int i = 0 ; i < m ; i ++){
        for(int j = 0 ; j < n ; j++){
            cin >> temp;
            aa.push_back(temp);
        }
        a.push_back(aa);
        aa.clear();
    }

    cout << "find num is :";
    cin >> num;

    res = findNum(a,num);

    cout <<"find result is : " << res;

    return 0;
}