有序矩阵查找练习题

传送门:http://www.nowcoder.com/courses/1/2/18

题目描述

现在有一个行和列都排好序的矩阵，请设计一个高效算法，快速查找矩阵中是否含有值x。

给定一个int矩阵mat，同时给定矩阵大小nxm及待查找的数x，请返回一个bool值，代表矩阵中是否存在x。所有矩阵中数字及x均为int范围内整数。保证n和m均小于等于1000。

测试样例：

[[1,2,3],[4,5,6],[7,8,9]],3,3,10

返回：false

时间复杂度可以做到O(M+N),额外空间复杂度为O(1);

AC代码

class Finder {
public:
    bool findX(vector<vector<int> > mat, int n, int m, int x) {
        int curRow=0,curCol=m-1;
        while (true) {
            if(mat[curRow][curCol]==x)
                return true;
            else{
                if(x>mat[curRow][curCol])
                    curRow++;
                else
                    curCol--;
                if(curRow>=n||curCol<0)
                    return false;
            }
        }
    }
};