剑指offer 1 二维数组中的查找 c++和python

题目

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

思路，一开始肯定是全循环，这样复杂度会比较高，题目给出数据是有序的，那么从右上角开始找，因为比右上角大的可能在他下面，比他小的可能在他左边。

c++版

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int row = 0,col = array[0].size()-1;
        while (row < array.size() && col >= 0){
            if(array[row][col] == target){
                return true;
            }
            else if(array[row][col] > target){
                --col;
            }
            else{
                ++row;
            }
        }
        return false;
    }
};

python版

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        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:
                row += 1
            else:
                col -= 1
        return False
        # write code here

或者利用python的 in，更加方便

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        for line in array:  #遍历每一行
            if target in line: #目标在这行里面吗
                return True
        return False
        # write code here