剑指offer--二维数组中的查找

二维数组中的查找

时间限制：1秒 空间限制：32768K
本题知识点：查找

题目描述

在一个二维数组中（每个一维数组的长度相同），每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。

思路

• 选取右上角元素

1. array[i][j] > target时，去掉整行，j–;
2. array[i][j] < target时，去掉整列，i++;
   

代码

1. 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;
    }
};

2. 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;
    }
}

3. 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