二维数组中的查找
时间限制:1秒 空间限制:32768K
本题知识点:查找
题目描述
在一个二维数组中(每个一维数组的长度相同),每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
思路
- 选取右上角元素
- array[i][j] > target时,去掉整行,j–;
- array[i][j] < target时,去掉整列,i++;
代码
- C++:
class Solution {
public:
bool Find(int target, vector<vector<int> > array) {
int row_len = (int)array.size();
int row = 0;
int col = (int)array[0].size()-1;
while(col >= 0 && row < row_len)
{
if(array[row][col] > target){
col--;
}else if(array[row][col] < target){
row++;
}else{
return true;
}
}
return false;
}
};
- Java代码:
public class Solution {
public boolean Find(int target, int [][] array) {
int row_len=array.length;
int row = 0;
int col = array[0].length-1;
while(col >= 0 && row < row_len)
{
if(array[row][col] > target){
col--;
}else if (array[row][col] < target){
row++;
}else{
return true;
}
}
return false;
}
}
- Python
# -*- coding:utf-8 -*-
class Solution:
# array 二维列表
def Find(self, target, array):
# write code here
row_len , col_len = len(array), len(array[0])
row = 0
col = col_len -1
while col >= 0 and row < row_len:
if array[row][col] == target:
return True
elif array[row][col] > target:
col -= 1
else:
row += 1
return False