有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的,请编写程序在这样的矩阵中查找某个数字是否存在。
要求:时间复杂度小于O(N);
解析:
这里,题目的要求是找到既可,所以不需要关注是否有重复数字的情况;如果有重复的数字出现,那么时间的复杂度也会变化,所以这里只需考虑找到就行了
答案:
#include<stdio.h>
int find_k(int arr[3][3], int* px, int* py, int k)
{
int x = 0;
int y = *py - 1;
while (x <= *py - 1 && y >= 0)
{
if (arr[x][y] < k)
{
x++;
}
else if (arr[x][y] > k)
{
y--;
}
else
{
*px = x;
*py = y;
return 1;
}
}
return 0;
}
int main()
{
int arr[3][3] = { 1,2,3,4,5,6,7,8,9 };
int x = 3;
int y = 3;
int k = 7;
int ret = find_k(arr, &x, &y, k);//也可以直接传值,然后在查找函数里直接打印结果即可
if (ret == 0)
printf("找不到\n");
else
printf("找到了,下标为 %d %d\n", x, y);
return 0;
}
看结果: