题目:
//杨氏矩阵
有一个二维数组.
数组的每行从左到右是递增的,每列从上到下是递增的.
在这样的数组中查找一个数字是否存在。
时间复杂度小于O(N);
数组:
1 2 3
2 3 4
3 4 5
1 3 4
2 4 5
4 5 6
分析:拿到这个题一开始的思路就是暴力解法,直接两个嵌套的for循环遍历二维数组,但是这样的时间复杂度就是O^2(n),题目要求的时间复杂度是O(n),所以这种方法排除。换个思路:首先观察这个二维数组中元素的排列规律,因为它是从左到右,从上到下依次递增的,这时候你有没有想到什么?和二分查找很像吧,同样是一个有序序列,只不过是二维数组。没关系,我们依然可以把二分查找的思想用到这里来。
从最右上角的那个数据开始查找,每次只需要将data与其进行比较,即可去掉一行(or一列),不断缩小比较范围。直到最后找到对应值。具体代码也不难,我们需要做的就是不断更新待查找行和列。
本题关键点:
(1)如何返回两个值(采用封装成结构体的方式或者传入指针带入返回值)
(2)二分查找思想的灵活运用
代码:
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
#include<stdlib.h>
#define ROW 3
#define COL 3
//因为要返回位置坐标(两个值),考虑到函数只能有一个返回值,所以这里采用了封装成结构体的方式
typedef struct Position
{
int x;
int y;
}Position;
Position FindNum(int(*arr)[COL],int data)
{
//从最右上角开始查找,如果找到就返回对应位置坐标,没找到对应的坐标则为-1,-1
Position p;
p.x = -1;
p.y = -1;
int row = 0;
int col = COL - 1;
while (row < ROW&&col >= 0)
{
if (data == arr[row][col])
{
p.x = row;
p.y = col;
return p;
}
else if (data>arr[row][col])
{
row++;
}
else
{
col--;
}
}
return p;
}
int main()
{
int arr[ROW][COL] = { { 1, 2, 3 }, { 4, 5, 6 },{ 7, 8, 9 } };
int key = 0;
printf("请输入要查找的数字:");
scanf("%d", &key);
Position p1=FindNum(arr, key);
if (p1.x != -1 && p1.y != -1)
{
printf("找到了,下标为%d %d\n", p1.x, p1.y);
}
else
{
printf("不存在\n");
}
system("pause");
return 0;
}
运行结果: