n*m的矩阵,行和列都递增有序,求是否出现target元素(面试题)

题目描述:给定一个n*m的矩阵,矩阵的每一行都是递增的,每一列也是递增的,给定一个元素target,问该target是否在矩阵中出现。

思路:先从最左下角的元素开始找,三种情况:

    1. 如果该元素大于target,说明target一定不会出现在最后一行中(因为最后一行也是递增的,其他的更大于target)

    2. 如果该元素小于target,说明target可能会出现在这一列的后面列中,因为最后元素是当前列当中最大的元素,前面的元素更小于target,所以只需要找该行的下一列

    3. 如果该元素等于target,直接返回true

代码如下

bool searchMatrix(vector<vector<int> >& matrix, int target) {
        int n = matrix.size();
        if (n == 0) return false;
        int m = matrix[0].size();
        int row = n - 1;
        int col = 0;
        while (row >= 0 && col < m) {
            if (matrix[row][col] > target) 
                row--;
            else if (matrix[row][col] < target) 
                col++;
            else
                return true;
        }
        return false;
    }
View Code

 

posted @ 2018-10-23 20:12  Howe_Young  阅读(842)  评论(0编辑  收藏  举报