题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路一:
-
从左下角来看矩阵,向上数字递减,向右数字递增
-
从左下角开始查找,当要查找数字比左下角数字大时,右移 ;要查找数字比左下角数字小时,上移
python
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row=len(array)-1
col=len(array[0])-1
i=row
j=0
while(i>=0 and j<=col):
if target<array[i][j]:
i-=1
elif target>array[i][j]:
j+=1
else:
return True
return False
public class Solution {
public boolean Find(int target, int [][] array) {
int m=array.length-1;
int n=array[0].length-1;
int i=m;
int j=0;
if (m!=-1 && n!=-1){
while(i>=0 && j<=m){
if(target<array[i][j]){
i--;
}
else if(target>array[i][j]){
j++;
}
else{
return true;
}
}
}
return false;
}
}
思路二:
*把每一行看成有序递增的数组,利用二分查找,通过遍历每一行得到答案。
*时间复杂度是nlogn
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row=len(array)
col=len(array[0])
for i in range(row):
low=0
high=len(array[i])-1
while(low<=high):
mid=(low+high)/2
if target>array[i][mid]:
low=mid+1
elif target<array[i][mid]:
high=mid-1
else:
return True
return False
public class Solution {
public boolean Find(int target, int [][] array) {
for(int i=0;i<=array.length-1;i++){
int low=0;
int high=array[0].length-1;
while(low<=high){
int mid=(low+high)/2;
if(target<array[i][mid]){
high=mid-1;
}
else if(target>array[i][mid]){
low=mid+1;
}
else return true;
}
}
return false;
}
}