剑指offer：二维数组中的查找（C++/Python）

题目描述

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

思路一： 暴力解法，将所有的都列出来进行比较。

C++

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int row = array.size();
        int col = array[0].size();
        for (int i = 0; i < row; i++)
        {
            for (int j = 0; j < col; j++)
            {
                if (array[i][j] == target)
                    return 1;
            }
        }
        return 0;
    }
};

思路二： 由于该矩阵的是从左到右递增，从上到下递增，故可以定位右上结点，从右上结点开始比较，若大于目标数字，则去掉当前行，若小于该数字，则去掉当前列，以此类推。时间复杂度比暴力法低。

C++

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

Python

# -*- coding:utf-8 -*-
class Solution:
    # array 二维列表
    def Find(self, target, array):
        # write code here
        row = len(array)
        col = len(array[0])
        a = 0
        b = col - 1
        while (a < row and b >= 0):
            if (array[a][b] == target):
                return True
            elif (array[a][b] < target):
                a += 1
            else:
                b -= 1
        return False