有一个数字矩阵,矩阵的每行从左到右是递增的,矩阵从上到下是递增的,请编写程序在这样的矩阵中查找某个数字是否存在。
要求:时间复杂度小于O(n)
通过题目描述可知该矩阵为一个有序二维数组(杨氏矩阵)
类似下图的矩阵:
思路分析:
该查找方法每次排除一行或者一列,查找次数大大降低,时间复杂度小于O(n)
根据分析结果实现代码:
int FindNum(int (*p)[3], int row, int col, int num) {
int r = 0;//行索引 从0开始
int c = col - 1;//列索引 从最后一列开始
while (r < row && c >= 0) {
if (p[r][c] < num) {
r++;//小于目标值行索引+1 跳过一行
}
else if (p[r][c] > num) {
c--;//大于目标值列索引-1 跳过一列
}
else {
return 1;//目标存在返回1
}
}
return 0;
}