牛客網:二維數組中的查找

題目描述

在一個二維數組中(每個一維數組的長度相同),每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個函數,輸入這樣的一個二維數組和一個整數,判斷數組中是否含有該整數。

解題思路:

從左下角開始比較,如果 target 比當前數組數字小,那麼就向上一行找,然後遍歷這行的,如果 target 比這行第一個數字大,那麼嘗試着向這行的右邊找,即可以找到;然後就是這個思路,一直循環尋找。

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size();
        int cols = array[0].size();
        int i = rows - 1, j = 0;
        while(i >= 0 && j < cols){
            if(target < array[i][j])	//如果 target 小於當前數組數字,那麼就向上找;
                i--;
            else if(target > array[i][j])	//如果 target 大於當前數組數字,那麼向右邊找;
                j++;
            else
                return true;
        }
        return false;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章