杨氏矩阵
题目:
有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的请编写程序在这样的矩阵中套找某个数字是否存在。
要求:时间复杂度小于O(N);
思路:
如果没有时间复杂度的限制,可以遍历整个数组,然后就可以判断是否存在想要的数字,但是这不符合题目的要求。
我们假设是3*3的矩阵,要查找数字7,x设为行数,y设为列数,我们先看右上角的3,对于3来说,3是这一行最大的数字,3是它所在列的最小的数字。如下图红圈的图所示。
如果要查找7,对于3来说,3这一行所在的数字都可以划去,如下图所示,因为3是这一行最大的数字,超过了3就说明这一行都没有比3大的。我们需要把行去掉。
如果要查找2,对于3来说,3这一列所在的数字都可以划去,如下图所示,因为3是这一列最小的数字,没有超过3就说明这一列都比3大。我们需要把列去掉。
int martix(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) {
y--;
}
else if (arr[x][y] < k) {
x++;
}
else {
*px = x;
*py = y;
return 1;
}
}
return 0;
}
int arr[3][3] = { 1,2,3,4,5,6,7,8,9 };
int k = 7; //需要查找的数字
int x = 3; //矩阵行数
int y = 3; //矩阵列数
int ret=martix(arr, &x, &y, k);
if (ret == 1) {
printf("找到了\n");
printf("下标为:%d %d", x, y);
}else{
printf("没找到\n");
}