有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的,请编写程序在这样的矩阵中查找某个数字是否存在。
要求:时间复杂度小于O(N);(若没有此要求,最容易想到并且实现的方式就时遍历二维数组)
从左到右递增,从上到下递增的矩阵如上均可。
按照此规则,可以得知每行最右侧的数最大,每列最下边的数最大
思路:将每行最大的元素与需要查找的k比较,若k比它小,就说明k有可能在此行,就将列下标减1;若k比它大,就说明k不在此行,就往下行寻找,将行下标加1;若k等于它,就说明找到了。
只有当行下标与列下标都在矩阵规定范围内才继续查找。
如上图:若想查找的k为5,先让arr[0][2]与5比较,arr[0][2] < 5,就将行下标加1,将arr[1][2]与5比较,arr[1][2] > 5,将列下标减1,将arr[1][1]与5比较,二者相等,则找到。
源代码实现如下:
#include<stdio.h>
int main()
{
int arr[3][3] = { 1,2,4,7,9,11,10,12,13 };
int k = 0;
scanf("%d", &k);
int i = 0;
int j = 2;
//flag是一个标志,初始化为0,表示找不到,若找到k,则会将k赋值为1
int flag = 0;
while (i < 3 && j >= 0)
{
//k大于某一行最大的元素,表示k不在此行,向下行查找
if (arr[i][j] < k)
{
i++;
}
//k小于某一行最大的元素,表示k有可能在此行,在此行中向前查找
else if (arr[i][j] > k)
{
j--;
}
//循环中最后一种情况就是找到了,此时打印下标并跳出循环
else
{
//将标志赋为1
flag = 1;
printf("找到k的下标为:%d,%d\n", i, j);
break;
}
}
//当矩阵中没有k时才会flag才仍会为0
if (flag == 0)
{
printf("找不到\n");
}
return 0;
}
运行结果如下: