1. 程式人生 > 其它 >T7——04. 二維陣列中的查詢

T7——04. 二維陣列中的查詢

在一個 n * m 的二維陣列中,每一行都按照從左到右遞增的順序排序,每一列都按照從上到下遞增的順序排序。請完成一個高效的函式,輸入這樣的一個二維陣列和一個整數,判斷陣列中是否含有該整數。

示例:

現有矩陣 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。

限制:

0 <= n <= 1000

0 <= m <= 1000

解題思想:

從右上角的元素開始比較 ,當 target 大於此元素則向下移動,當 target 小於此元素則向左移動,當 target 等於此元素則返回 true,當 target 沒找到相同元素則返回 false。

程式碼:

let twoM: number[][] = [ [ 1, 4, 7, 11, 15 ], [ 2, 5, 8, 12, 19 ], [ 3, 6, 9, 16, 22 ], [ 10, 13, 14, 17, 24 ] ];
function findNumberIn2DArray(matrix: number[][], target: number): any {
    
if (matrix.length === 0) { return false; } const row = matrix.length - 1; const list = matrix[0].length - 1; var x = list; var y = 0; while (y <= row && x >= 0) { if (target < matrix[y][x]) { x--; } else if (target > matrix[y][x]) { y
++; } else if (target == matrix[y][x]) { return true; } } return false; } console.log(findNumberIn2DArray(twoM, 5));