题目描述:
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
解题思路:
- 利用二维数组由上到下,由左到右递增的规律,那么选取右上角的元素与target进行比较;
- 当target小于右上角的元素时,那么它将小于整个列,因此,target必定在此元素所在列的左边, 即列号--;
- 当target大于右上角的元素时,那么它将大于整个行,因此,target必定在此元素所在行的下边, 即行号++;
- 如果相等,查找结束。
java实现:
public class Solution {
public boolean Find(int target, int [][] array) {
int i = 0;
int j = array[0].length-1;
while(i<array.length && j>=0) {
if (array[i][j] == target) {
return true;
} else if (array[i][j] < target) {
i++;
} else {
j--;
}
}
return false;
}
}
python实现:
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
if not array:
return False
row = 0
col = len(array[0]) - 1
while row < len(array) and col >= 0:
if array[row][col] == target:
return True
elif array[row][col] > target:
col -= 1
else:
row += 1
return False