C问题---在杨氏矩阵中查找

Posted 2021-06-26 4nc414g0n

题目

1. 有一个数字矩阵，矩阵的每行从左到右是递增的，矩阵从上到下是递增的，请编写程序在这样的矩阵中查找某个数字是否存在。
2. 要求：时间复杂度小于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