04 二维数组中的查找

题目

在一个二维数组中，每一行都按照从左到右递增的顺序排序，每一列都按照从上到下递增的顺序排序。请完成一个函数，输入这样的一个二维数组和一个整数，判断数组中是否含有该整数。
例如下面的二维数组就是每行、每列都递增排序。如果在这个数组中查找数字7，则返回true；如果查找数字5，由于数组不含有该数字，则返回false。

牛客网 OJ
AcWing OJ

C 语言 题解

首先选取数组中右上角的数字：

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

bool Find(int *matrix, int rows, int cols, int num)
{	
	bool found = false;

	if (matrix != nullptr && rows > 0 && cols > 0)
	{
		int row = 0;
		int col = cols - 1;

		while (row < rows && col >= 0)
		{
			if (matrix[row * cols + col] == num)
			{
				found = true;
				break;
			}
			else if (matrix[row * cols + col] > num)
			{
				col -= 1;
			}
			else
			{
				row += 1;
			}
		}
	}	

	return found;
}

注意:

1. col >= 0 这样才能够取到第一列的值；
2. 也可以按照从左下角开始查找的思路进行查找。

C++ 题解

class Solution {
public:
    bool Find(int target, vector<vector<int> > array) {
        int rowCount = array.size();
        int colCount = array[0].size();
        int i,j;
        for(i=0,j=colCount-1 ;i<rowCount && j>=0 ;)
        {
            if(target == array[i][j])
                return true;
            if(target < array[i][j])
            {
                j--;
                continue;
            }
            if(target > array[i][j])
            {
                i++;
                continue;
            }
        }
        return false;  
        
    }
};

python 题解

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