剑指Offer（一）：二维数组中的查找

摘要

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

一、前言

本系列文章为《剑指Offer》刷题笔记。

刷题平台：牛客网

书籍下载：共享资源

二、题目

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

1、思路

首先选取数组中右上角的数字。如果该数字等于要查找的数字，查找过程结束；如果该数字大于要查找的数组，剔除这个数字所在的列；如果该数字小于要查找的数字，剔除这个数字所在的行。也就是说如果要查找的数字不在数组的右上角，则每一次都在数组的查找范围中剔除一行或者一列，这样每一步都可以缩小查找的范围，直到找到要查找的数字，或者查找范围为空。

2、举例

如果在一个二维数组中找到数字7，则返回true，如果没有找到，则返回false。

查找过程如下：

3、编程实现

C++：

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23

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

Python2.7：

Python

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

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

java:

public static boolean findKey(int [][] A,int key)
    {
        if(A == null) return false;
        int rows = A.length;
        int cols = A[rows-1].length;
        rows = 0;
        while(rows<A.length - 1 && cols>0){
            if(A[rows][cols-1] == key)
                return true;
            if(A[rows][cols-1] > key)
            {
                cols--;
                continue;
            }
            if(A[rows][cols-1] < key){
                rows++;
                continue;
            }
        }
        return false;
    }