二维数组查找元素

二维数组查找元素

题目描述
二维数组查找元素_第1张图片
解题思路:有两种方法,暴力枚举O(NM)和线性查找O(N+M)

这里只讲解线性查找的方法。暴力的话就直接遍历,在实际面试中是不可取的。

那么怎样去分析这种升序二维数组查找元素的通法。

算法流程:

1.首先找到最右上角的元素,array[ i ][ j ],要找的target可以与其做比较
对二维数组进行分析可以发现每一行中最右边的元素是最大的,严格单调增

2.target比最右边这个元素要小,那么j-- 就是列减少我们倒着遍历这一行每个元素,直到找到最终的target

3.target比最右边这个元素还大,这一行就肯定不存在我们要找的target了,这时候我们要在下一行去找 i++

4.target比array[ i ][ j ]小的话,我们就在 i+1 这一行去找,不然再到 i+2 行去找…

代码:

class Solution {
public:
    bool findNumberIn2DArray(vector<vector<int>>& matrix, int target) {
    if(matrix.empty() || matrix[0].empty()) return false;
    int i=0, j=matrix[0].size()-1;//j相当于得到列数
    while(i<matrix.size()&&j>=0){
        if(matrix[i][j] == target) return true;
        else if(matrix[i][j] > target) j--;//如果目标值比当前值低 那就可能在这一行 
        else i++; //如果目标值比当前值高 那就从下一行开始遍历
    }
    return false;
    }
};

你可能感兴趣的:(leetcode,算法,矩阵)