n*m的矩陣,行和列都遞增有序,求是否出現target元素(面試題)

題目描述:給定一個n*m的矩陣,矩陣的每一行都是遞增的,每一列也是遞增的,給定一個元素target,問該target是否在矩陣中出現。ide

思路:先從最左下角的元素開始找,三種狀況:spa

    1. 若是該元素大於target,說明target必定不會出如今最後一行中(由於最後一行也是遞增的,其餘的更大於target)code

    2. 若是該元素小於target,說明target可能會出如今這一列的後面列中,由於最後元素是當前列當中最大的元素,前面的元素更小於target,因此只須要找該行的下一列blog

    3. 若是該元素等於target,直接返回trueget

代碼以下event

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
相關文章
相關標籤/搜索