在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的个二维数组和一个整数,判断数组中是否含有该整数。
选取数组中右上角的数字(也可以选取左下角的数字)。如果该数字等于要查找的数字,则査找过程结束;如果该数字大于要查找的数字,则剔除这个数字所在的列;如果该数字小于要查找的数字,,则剔除这个数字所在的行。也就是说,如果要査找的数字不在数组的右上角,则每一次都在数组的査找范围中剔除一行或者一列,这样每一步都可以缩小查找的范围,直到找到要查找的数字,或者査找范围为空。
func find(matrix: [[Int]], number: Int) -> Bool {
guard !matrix.isEmpty,
let columns = matrix.max(by: { $0.count < $1.count })?.count
else { return false }
var b = true
matrix.forEach { numbers in
if numbers.count != columns {
b = false
}
}
if !b {
return false
}
let rows = matrix.count
var row = 0
var column = columns - 1
while row < rows, column >= 0 {
if matrix[row][column] == number {
return true
} else if matrix[row][column] > number {
column -= 1
} else {
row += 1
}
}
return false
}
摘抄资料:《剑指offer》
网友评论