这道算法题可以使用两种方法来解决:暴力搜索和二分查找。
1.暴力搜索
暴力搜索的思路是从二维数组的左下角或右上角开始,逐行或逐列搜索,直到找到目标元素或者搜索到数组边缘为止。
定义两个变量 i 和 j 来表示当前位置的行和列,初始值为左下角的坐标。如果当前位置的值等于目标值,则返回 true。如果当前位置的值小于目标值,则说明目标值位于当前位置的右侧,我们将列 j 加 1,继续搜索。如果当前位置的值大于目标值,则说明目标值位于当前位置的上方,我们将行 i 减 1,继续搜索。如果搜索结束后还没有找到目标值,则说明二维数组中不包含该值,返回 false。
function searchIn2DArray(matrix, target) {
let i = matrix.length - 1, j = 0;
while (i >= 0 && j < matrix[0].length) {
if (matrix[i][j] === target) {
return true;
} else if (matrix[i][j] < target) {
j++;
} else {
i--;
}
}
return false;
}
2.二分查找
由于二维数组的每行和每列都已经排序,因此可以将二维数组视为一个一维数组,使用二分查找算法来查找目标元素。具体实现方法是将二维数组的行列坐标转化为一维数组的下标,然后进行二分查找。
首先计算出二维数组的行数 row 和列数 col。然后使用二分查找算法来查找目标元素。我们定义一个变量 left 和 right 分别表示一维数组的左右边界。在每次循环中,计算 mid,然后将二维数组的行列坐标转化为一维数组的下标。比如对于二维数组中的 matrix[i][j],对应的一维数组的下标为 i * col + j。然后判断当前元素是否等于目标值,如果等于直接返回 true;如果当前元素小于目标值,则说明目标值在右侧,将左边界 left 调整为 mid + 1;如果当前元素大于目标值,则说明目标值在左侧,将右边界 right 调整为 mid - 1。如果整个循环结束后还没有找到目标值,则说明二维数组中不包含该值,返回 false。
function searchIn2DArray(matrix, target) {
const row = matrix.length;
if (row === 0) {
return false;
}
const col = matrix[0].length;
let left = 0, right = row * col - 1;
while (left <= right) {
const mid = Math.floor((left + right) / 2);
const num = matrix[Math.floor(mid / col)][mid % col];
if (num === target) {
return true;
} else if (num < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return false;
}