240. Search a 2D Matrix II

題目

240. Search a 2D Matrix II

解題思路

這道題與74. Search a 2D Matrix的不同在於這道題目的矩陣是左上角的數字一定比右下角的大。

這道題我是看discussion的題解做的，先把初始位置定在右上角，這樣向下移就是增大，向左移就是減小，即有增大減小兩個不同方向。

如果把初始位置定在左上角，向右移和向下移都是增大，即兩個方向都是增大，無法確定是要哪個方向才能找到target。

具體代碼

class Solution {
public:
    bool searchMatrix(vector<vector<int>>& matrix, int target) {
        int m = matrix.size();
        int n = m > 0 ? matrix[0].size() : 0;
        if (n <= 0) {
            return false;
        }
        // 把初始位置定在右上角
        int i = 0, j = n - 1;
        while (i < m && j >= 0) {
            if (matrix[i][j] == target) {
                return true;
            }
            else if (matrix[i][j] < target) {
                i++; // 下移增大
            }
            else {
                j--;// 左移減小
            }
        }
        return false;
    }
};