问题:在一个m行n列二维数组中,每一行都按照从左到右递增的顺序排序,每一列都按照从上到下递增的顺序排序。请完成一个函数,输入这样的一个二维数组和一个整数,判断数组中是否含有该整数。
例如:下面的二维数组就是每行、每列都递增有序。如果在这个数组中查找数字6,则返回true;如果查找数字5,由于数组不含该数字,则返回false。
思路:根据数组的特点,从右上角开始按先行后列的策略进行查找,如果当前值大于要查找的数值,则向下(按列)查找;否则向左(按行)查找。直到找个这个数或者没有找到这个数字。例外一种查找策略是从左下角开始查找,与上述策略恰好相反。
算法实现——时间复杂度为O(m+n),空间复杂度为O(1).
/*
*
* 杨氏矩阵的查找
*
* Created on: 2014-4-7 22:33:35
* Author: danDingCongRong
*
*/
#include<iostream>
using namespace std;
//杨氏矩阵的查找
int youngMatrixSearch(int** matrix, int m, int n, int findValue) {
if (NULL == matrix || 0 >= m || 0 >= n) {
return 0;
}
for (int i = 0; i < m; ++i) {
for (int j = n - 1; j >= 0; --j) {
if (matrix[i][j] > findValue) {
continue;
} else if (matrix[i][j] < findValue) {
break;
} else {
return matrix[i][j];
}
}
}
return 0;
}
int main() {
int m = 0, n = 0;
cout << "输入矩阵的行数和列数:" << endl;
cin>>m>>n;
int** x = new int*[m];
for (int i = 0; i < m; ++i) {
x[i] = new int[n];
}
cout << "输入这个矩阵:" << endl;
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
cin >> x[i][j];
}
}
int value = 0;
cout << "输入要查找的数字:" << endl;
while (cin >> value) {
cout << youngMatrixSearch(x, m, n, value) << endl;
cout << "输入要查找的数字:" << endl;
}
for (int i = 0; i < m; ++i) {
delete x[i];
}
delete x;
return 0;
}
注:部分内容参照自剑指offer