4.二维数组中的查找
题目:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序,请完成一个函数,输入这样的一个二维数组和一个整数,判断该数组中是否含有该整数。
暴力解法(鄙人的最爱),时间复杂度
O
(
n
2
)
O(n^2)
O ( n 2 ) ,空间复杂度
O
(
1
)
O(1)
O ( 1 ) 。
class solution1 {
public boolean find ( int [ ] [ ] array, int target) {
if ( array == null || array. length == 0 ) return false ;
for ( int i = 0 ; i < array. length; i++ ) {
for ( int j = 0 ; j < array[ i] . length; j++ ) {
if ( array[ i] [ j] == target) return true ;
}
}
return false ;
}
}
双指针法,我们可以根据二维数组的结构进行查找,省去不必要的遍历,由于行从左向右递增,列从上到下递增,我们直接选取第一行最后一列,判断其与目标值的大小,如果大了则查找下一行行最后一列,如果小了则查找本行的前一列。时间复杂度
O
(
m
n
)
O(mn)
O ( m n ) ,空间复杂度
O
(
1
)
O(1)
O ( 1 ) 。
class solution2 {
public boolean find ( int [ ] [ ] array, int target) {
if ( array == null || array. length == 0 ) return false ;
int row = 0 ;
int col = array[ 0 ] . length - 1 ;
while ( row < array. length && col >= 0 ) {
if ( array[ row] [ col] == target) return true ;
if ( array[ row] [ col] > target) {
col-- ;
} else {
row++ ;
}
}
return false ;
}
}
使用二分法,使用二分查找的思想,时间复杂度
O
(
l
o
g
m
n
)
O(logmn)
O ( l o g m n ) ,空间复杂度
O
(
1
)
O(1)
O ( 1 ) 。
class solution3 {
public boolean find ( int [ ] [ ] array, int target) {
if ( array == null || array. length == 0 ) return false ;
int start = 0 ;
int end = array. length * array[ 0 ] . length - 1 ;
int col = array[ 0 ] . length;
while ( start <= end) {
int mid = ( start + end) / 2 ;
int val = array[ mid / col] [ mid % col] ;
if ( val == target) {
return true ;
} else if ( value < target) {
start = mid + 1 ;
} else {
end = mid - 1 ;
}
}
return false ;
}
}