题目
- 有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的,请编写程序在这样的矩阵中查找某个数字是否存在。
- 要求:时间复杂度小于O(N);
#include <stdio.h>
#include <assert.h>
int find_num(int arr[3][3],int *px,int *py,int k)
{
int x=0;
int y=*py-1;
while(x<*px&&y>=0)
{
if(arr[x][y]<k)
{
x++;
}
else if(arr[x][y]>k)
{
y--;
}
else
{
return 1;
}
}
return 0;
}
int main()
{
int arr[3][3]={1,2,3,4,5,6,7,8,9};
int k=7;
int x=3;
int y=3;
int ret=find_num(arr,&x,&y,k);
if(ret==1)
{
printf("found\n");
}
else{
printf("not found\n");
}
return 0;
}
分析:
例:
1 2 3
4 5 6
7 8 9
从右上角开始比较,比k小就 行+1
再右上角开始比较,比k大就 列-1
最后如果找到就返回1