1. 程式人生 > 實用技巧 >15. 二維陣列中的查詢

15. 二維陣列中的查詢

題目描述

在一個二維陣列中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。

請完成一個函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。

樣例

輸入陣列:

[
  [1,2,8,9],
  [2,4,9,12],
  [4,7,10,13],
  [6,8,11,15]
]

如果輸入查詢數值為7,則返回true,

如果輸入查詢數值為5,則返回false。

思路

  由題目的描述可知,輸入的是一個楊式矩陣(行列都是遞增的矩陣),要查詢某個數,那麼就從對角線上的兩個數(右上角或左下角)開始,因為右上角的數是該行最大該列最小的數,而左下角的數是該行最小該列最大的數,以右上角的數為例,若要查詢的數小於這個數,那麼就往左移動一格繼續查詢,若大於這個數,那麼就往下移動一格繼續查詢,這樣就能在最短時間內判斷出矩陣中是否含有我們要查詢的數.

  需要注意的是,輸入的二維陣列可能為空,所以需要對此先行判斷.

 1 class Solution {
 2 public:
 3     bool searchArray(vector<vector<int>> array, int target) {
 4         int n=array.size();
 5         if(!n) return false;//若二維陣列為空
 6         int m=array[0].size();
 7         int i=0, j=m-1;
 8         while(1) {
 9             if
(i<0 || i>=n || j<0 || j>=m) break; 10 if(array[i][j]==target) return true; 11 if(array[i][j]<target) i++; 12 else if(array[i][j]>target) j--; 13 } 14 return false; 15 } 16 };