題目描述

編寫一個高效的演算法來搜尋 m x n 矩陣 matrix 中的一個目標值 target。該矩陣具有以下特性：

• 每行的元素從左到右升序排列。
• 每列的元素從上到下升序排列。

示例:

現有矩陣 matrix 如下：

[
  [1,   4,  7, 11, 15],
  [2,   5,  8, 12, 19],
  [3,   6,  9, 16, 22],
  [10, 13, 14, 17, 24],
  [18, 21, 23, 26, 30]
]

給定 target = 5，返回 true

給定 target = 20，返回 false。

解題思路

從右上角開始比較，若當前元素小於target，說明此行沒有目標，所以向下移一行繼續尋找；若當前元素大於target，說明此列向下沒有目標，所以向左移一列，直到找到目標值。

程式碼

 1 class Solution {
 2 public:
 3     bool searchMatrix(vector<vector<int>>& matrix, int target) {
 4         if
 
(matrix.empty()) return false;
 5         int rows = matrix.size(), cols = matrix[0].size();
 6         int i = 0, j = cols - 1;
 7         while(i >= 0 && i < rows && j >= 0 && j < cols){
 8             if(matrix[i][j] == target) return true;
 9             else if(matrix[i][j] > target) j--;
 
10             else i++;
11         }
12         return false;
13     }
14 };