15. 二維陣列中的查詢
阿新 • • 發佈:2020-09-14
題目描述
在一個二維陣列中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。
請完成一個函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。
樣例
輸入陣列:
[
[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 };