編寫一個高效的算法來搜索 m x n 矩陣 matrix 中的一個目標值 target。該矩陣具有以下特性:
- 每行的元素從左到右升序排列。
- 每列的元素從上到下升序排列。
示例:
現有矩陣 matrix 如下:
[
[1, 4, 7, 11, 15],
[2, 5, 8, 12, 19],
[3, 6, 9, 16, 22],
[10, 13, 14, 17, 24],
[18, 21, 23, 26, 30]
]
給定 target = 5
,返回 true
。
給定 target = 20
,返回 false
。
首先選取矩陣右上角數字,如果該數字等於要查找的數字,返回true;如果該數字大於要查找的數字,去掉此數字所在列;如果該數字小於要查找的數字,則去掉該數字所在行。重複上述過程直到找到要查找的數字,或者查找範圍爲空。
bool searchMatrix(int** matrix, int matrixRowSize, int matrixColSize, int target) {
int i=0,j=matrixColSize-1;
while(i<matrixRowSize&&j>=0)
{
if(matrix[i][j]==target)
return true;
else if(matrix[i][j]<target)
i++;
else
j--;
}
return false;
}
int main()
{
int a[][5]={{1,4,7,11,15},{2,5,8,12,19},{3,6,9,16,22},{10,13,14,17,24},{18,21,23,26,30}};
int *p[5];
for(int i=0;i<5;i++)
p[i]=a[i];
printf("%d",searchMatrix(p,5,5,5));
system("pause");
return 0;
}