剑指Offer 面试题3 二维数组中查找
参加鹅厂实习生笔试,在算法上被虐惨,做项目花了大部分时间,算法能力没能提高,迫切需要提高算法能力。遂收集各种面试题,刷刷算法。
本文参考《剑指offer 名企面试官精讲典型编程题》面试题3。
题目3:在一个二维数组中,每一行都按照从左到右递增的顺序排序,每一列按照从上到下递增的顺序排序。请完成一个函数,输入一个这样的二维数组和整数,判断数组中是否含有该整数。
例如下面的二维数组就是每行每列递增排序。如果在数组中查询7,则返回true;如果查找数字14,由于数组中不包含14,则返回false。
解决方法分析:
首先我们选取二维数组左下角的数字8,由于8大于7,并且8还是第四行的第一个数,因此数字7不可能出现在8所在的行,于是我们把这一行从需要考虑的区域中剔除,之后需要分析的只剩下3行。在剩下的矩阵中,位于左下角的数字是4,4小于7,可以把4所在的列从需要考虑的区域中剔除。接下来只需要分析剩下的3行3列即可。同理,可以照之前处理方式继续处理。
处理过程如下图所示:
注:矩阵灰色区域为下一步要查找的范围,绿色为所要查找的数字。
整个算法的逻辑:
首先选取数组左下角的数字。如果该数字等于要查找的数字,查找过程结束;如果该数字大于要查找的数字,剔除这个数字所在的行;如果该数字小于要查找的数字,剔除这个数字所在的列。也就是说,每次剔除一行或一列来缩小查找范围,直至找到要查找的数字,或者查找对应数字失败!
当然除了选择左下角的数字外,也可以选择右上角的数字,查找方法类似。
下面我以Java为编程语言,展示上述算法的处理过程。
package array;
/**
* @author WuPing
* @version 2016年4月7日 下午7:11:44
*/
public class FindInPartiallySortedMatrix {
public static boolean FindNumber(int[] matrix, int rows, int columns,
int number) {
boolean found = false;
if (matrix != null && rows > 0 && columns > 0) {
//以二维数组左下角的数为查找比对标准
int row = rows - 1;
int column = 0;
while (row >= 0 && column < columns) {
if (matrix[row * columns + column] == number) {
found = true;
break;
} else if (matrix[row * columns + column] > number) {
--row;
} else {
++column;
}
}
}
return found;
}
public static void Test(String testName, int[] matrix, int rows,
int columns, int number, boolean expected) {
if(testName != null) {
System.out.print(testName + " begins: ");
boolean result = FindNumber(matrix, rows, columns, number);
if(result == expected) {
System.out.print("Passed.");
System.out.println();
}
else {
System.out.print("Failed.");
System.out.println();
}
}
}
public static void main(String[] args) {
//此处以一维数组加行数和列数的形式表示二维数组
int[] matrix = {1, 3, 6, 9, 2, 5, 7, 11, 4, 10, 13, 16, 8, 12, 15, 18};
//要找的数在二维数组中,但不在左上角或右下角
Test("Test1", matrix, 4, 4, 7, true);
//要找的数不在二维数组中,但在数组范围内
Test("Test2", matrix, 4, 4, 14, false);
//要找的数在二维数组左上角
Test("Test3", matrix, 4, 4, 1, true);
//要找的数在二维数组右下角
Test("Test4", matrix, 4, 4, 18, true);
//要找的数在二维数组范围下界外
Test("Test5", matrix, 4, 4, 0, false);
//要找的数在二维数组范围上界外
Test("Test6", matrix, 4, 4, 20, true);
//传递的二维数组为空
Test("Test7", null, 4, 4, 20, false);
}
}
代码执行结果如下: