杨氏矩阵

刚开始我的思路，只能局限于正方形矩阵中，在矩形中并不适用

#include <stdio.h>
#include <windows.h>
#define ROW 3
#define COL 3
#pragma warning(disable:4996)

/*杨氏矩阵 
有一个二维数组. 
数组的每行从左到右是递增的，每列从上到下是递增的. 
在这样的数组中查找一个数字是否存在。 
时间复杂度小于O(N); 
*/

int FindNum(int a[][3],int row,int col,int key){
	int flag = 0;
	int m = 0;
	int n = col - 1;
	int start = 0;
	start = a[m][n];
	while (m<row && n>=0){
		if (key == start){
			return flag;
			break;
		}
		else if (key > start){
				m++;
			}
			else{
				n--;
			}
	}
	return flag;
}

int main(){
	int a[ROW][COL] = { 1,2,3,2,3,4,3,4,5 };
	int t;
	printf("Enter the number of you want to find:\n");
	scanf("%d",&t);
	printf("%d\n", FindNum(a, ROW, COL, t));

	system("pause");
	return 0;
}