剑指offer—JZ4 二维数组中的查找
题目描述:
杨氏矩阵,是对组合表示理论和舒伯特演算很有用的工具。它提供了一种方便的方式来描述对称和一般线性群的群表示,并研究它们的性质。有一个二维数组. 数组的每行从左到右是递增的,每列从上到下是递增的. 在这样的数组中查找一个数字是否存在。 时间复杂度小于O(N);
而本题二维数组矩阵则刚好满足杨氏矩阵的特征
矩阵每一行元素从左到右依次递增;每一列从上到下依次递增
因此有了这种算法:
线性搜索算法
我们仔细分析,不难发现,对于杨氏矩阵老说,右上角和左下角的元素是有特点的。右上角的元素是一行中最大的,一列中最小的。左下角的元素是一行中最小的,是一列中最大的。所以我们可以从右上角或者左下角开始查找。比如:从右上角开始查找的时候,右上角的元素比我们要查找元素小,我们就可以去掉右上角元素所在的这一行;右上角的元素比我们要查找的元素大,我们就可以去掉右上角元素所在的这一列。然后依然找右上角的元素继续和要查找的元素与比较。这样每一次比较去掉一行或者去掉一列。这个查找效率是高于遍历数组元素的,所以时间复杂度是小于O(N),也满足题目要求。
杨氏矩阵的特点是,最右上角看,下方的数据一定大、左方的数据一定小;对于左下角,上方的数据一定小,右方的数据一定大。
比如待搜索元素为47,右上角元素为9,那么右上角元素所在行删除(因为,行逐渐递增),同理11小于47因此又删除一行,然后35小于47,因此删除一行…最终查找的路线为下图中红色
最差的情况下,查找从右上角直到左下角对于x*y矩阵,时间复杂度O(x+y)。
因此最终的代码如下:
#include <stdio.h>
int findnum(int a[][3], int x, int y, int f) //第一个参数的类型需要调整
{
int i = 0, j = y - 1; //从右上角开始遍历
while (j >= 0 && i < x)
{
if (a[i][j] < f) //比我大就向下
{
i++;
}
else if (a[i][j] > f) //比我小就向左
{
j--;
}
else
{
return 1;
}
}
return 0;
}
int main()
{
int a[][3] = { {1, 3, 5},
{3, 5, 7},
{5, 7, 9} }; //一个示例
if (findnum(a, 3, 3, 2))
{
printf("It has been found!\n");
}
else
{
printf("It hasn't been found!\n");
}
return 0;
}