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

题目描述

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

 

分析：

自然容易想到二分法，但二分法不是最优解，利用元素特殊的排列进行删选（每次选取右上角或者左下角的数字）

以下列二维数组为例 ，从9开始，如果9>target，那么第四列整列都比target大，可以删掉，col-1；如果9<target，那么第一行整行都比target小，可以删掉，rol+1。以此类推直到找到等于target的数。

 

代码：

c++版本：

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rows = array.size();
        int cols = array[0].size();
        if(!array.empty() && rows > 0 && cols > 0){
            int row = 0;
            int col = cols - 1;
            while(row < rows && 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):
        # write code here
        row =0
        col = len(array[0])-1
        while col>=0 and row<len(array):
            if target > array[row][col]:
                row += 1
            elif target < array[row][col]:
                col -= 1
            else:
                return True