《剑指offer》面试题 4：二维数组中的查找（C++实现）

题目

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

代码

#pragma once
#include <iostream>
using namespace std;

/**
 * 从具体问题入手，根据二维数组的行号和列号计算出相对于其首地址的偏移量，找出相应元素
 * 可用右上角数字或左下角数字作为判断点
 */
bool findInMatrix(int* matrix, int rows, int columns, int number)
{
	bool found = false;
	if (matrix != nullptr && rows > 0 && columns > 0)
	{
		int row = 0;
		int column = columns - 1;	// 右上角数字
		while (row < rows && column >=0)
		{
			if (matrix[row * columns + column] == number)
			{
				found = true;
				break;
			}
			else if (matrix[row * columns + column] > number)
				--column;
			else
				++row;
		}
	}

	return found;
}

总结

从右上角（左下角）查找数字，每次删除一行或一列以缩小查找范围，直至找到数字。二维数组表示：matrix[row * columns + column] 。