剑指offer——在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。时间复杂度小于O(n)。
分析:先把要查找的数字和右上角的作比较,如果要查找的数字大,由于数组每行每列都是递增的,那么剔除第一行元素。如果要查找的数字小,则剔除最后一列。一次剔除后将要查找数字和下一个右上角元素作比较,此次类推,直至找到所找数组返回它,或未找到返回0。同理,也可以从左下角开始比较。但不能从左上角或右下角作比较,因为无法剔除。
图示过程为二维数组从右上角查找数字7:
#include<stdio.h>
#include<stdlib.h>
#include<assert.h>
int Findnum1(int arr[][4], int row, int col, int num)
{
int i = 0;
int j = col - 1;
if (num > arr[row-1][col-1] || num < arr[0][0]){
return 0;
}
while (i < row - 1 && j >= 0){
if (arr[i][j] > num){
j--;
}
else if (arr[i][j] < num){
i++;
}
else{
return num;
}
}
return 0;
}
int Findnum2(int arr[][4], int num){
int i = 3;
int j = 0;
if (num > arr[3][3] || num < arr[0][0]){
return 0;
}
while (i >= 0 && j <= 3){
if (num > arr[i][j]){
j++;
}
else if (num < arr[i][j]){
i--;
}
else{
return num;
}
}
return 0;
}
int main()
{
int a[4][4] = { 1, 2, 8, 9, 2, 4, 9, 12, 4, 7, 10, 13, 6, 8, 11, 15 };
int ret = 0;
//ret = Findnum1(a,4,4, 80);//右上角开始查找
ret = Findnum2(a, 8);//左下角开始查找
if (ret){
printf("找到数字%d\n", ret);
}
else{
printf("找不到\n");
}
system("pause");
return 0;
}