1. 程式人生 > >[CareerCup] 11.6 Search a 2D Matrix 搜尋一個二維矩陣

[CareerCup] 11.6 Search a 2D Matrix 搜尋一個二維矩陣

11.6 Given an M x N matrix in which each row and each column is sorted in ascending order, write a method to find an element.

class Solution {
public:
    bool findElement(vector<vector<int> > &matrix, int elem) {
        if (matrix.empty() || matrix[0].empty()) return false
; int row = 0, col = matrix[0].size() - 1; while (row < matrix.size() && col >= 0) { if (matrix[row][col] == elem) return true; else if (matrix[row][col] < elem) ++row; else --col; } return false; } };