题目名称:
杨氏矩阵
题目内容:
有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的,请编写程序在这样的矩阵中查找某个数字是否存在。
要求:时间复杂度小于O(N);
思路一: 数组行列递增,是有规律的,所以可以遍历数组寻找数字,但是这题有时间复杂度的要求,遍历数组时间复杂度是O(N*2),不满足题意。
思路二: 如图所示,矩阵右上角的数字5是它这一行里最大的,是它这一列里的最小的。
矩阵左下角的数字21是它这一行里最小的,是它这一列里的最大的。
借助这个结论,可以发现,比如要找数字7,它比右上角的数字5大,所以可以排除第一行;比如要找数字23,它比左下角的21大,所以可以排除第一列。
代码如下:
int search(int arr[3][3], int r, int c, int k) { int x = 0; int y = c - 1; while (x < r && c >= 0) { if (k > arr[x][y]) x++; else if (k < arr[x][y]) y--; else return 1; } return 0; } int main() { int arr[3][3] = {1,2,3,4,5,6,7,8,9}; int k = 0; scanf("%d", &k); int ret = search(arr,3,3, k); if (ret == 1) printf("YES\n"); else printf("NO\n"); return 0; }