一、题目:
在杨氏矩阵中查找一个元素,若存在则返回1,不存在则返回0.
二、介绍:
杨氏矩阵是一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的。
例如:
三、分析:
对于杨氏矩阵来说,右上角和左下角的元素是有特点的。右上角的元素是一行中最大的,因为每一行是从左往右递增的,同时还是一列中最小的,因为从上往下是递增的。左下角的元素是一行中最小的,是一列中最大的。所以我们可以从右上角或者左下角开始查找。比如:从右上角开始查找的时候,右上角的元素比我们要查找元素小,我们就可以去掉右上角元素所在的这一行;右上角的元素比我们要查找的元素大,我们就可以去掉右上角元素所在的这一列。然后依然找右上角的元素继续和要查找的元素与比较。这样每一次比较去掉一行或者去掉一列。这个查找效率是高于遍历数组元素的,所以时间复杂度是小于O(N),也满足题目要求。
int find_key(int arr[][3], int x, int y, int k)
{
int i = 0, j = y - 1;
while (j>=0&&i<x)
{
if (arr[i][j] > k)//如果大于目标元素
{
j--;//那么就左移
}
else if (arr[i][j] < k)//如果小于目标元素
{
i++;//那么久下移
}
else//如果相等
{
return 1;
}
}
return 0;
}
int main()
{
int arr[][3] = { {1, 2, 3},
{4, 5, 6},
{7, 8, 9} };
int k = 0;
scanf("%d", &k);
if (find_key(arr, 3, 3, k))
{
printf("找到了目标元素\n");
}
else
{
printf("没有找到目标元素\n");
}
return 0;
}
注:记得加上头文件!