给出一个矩阵,有下列特点:每行严格递增,每列严格递增,矩阵内没有重复元素
如:1 2 4 12
3 5 9 15
6 7 10 16
8 13 18 20
求K是否在矩阵中,写出思路和算法。
基本思路:因为各行各列都是单调递增的,所以对角线元素也是单调递增的,首先通过在对角线元素上运用二分查找找到所对应的的区间,然后在对于坐上角元素(start, start)至(start,n-1)、(start, start)至(n-1,start)以及(0,end)至(end,end)、(end,0)至(end,end)分别进行4次二分查找运算便能得出结果,总的时间复杂度为log(n)。以下为源码:
#include <cstdlib>
#include <iostream>
using namespace std;
int matrix[4][4] = {{1,2,4,12}, {3,5,9,15}, {6,7,10,16},{8,13,18,20}};
bool binarySearch(int n, int value)
{
int start = 0;
int end = n-1;
int middle;
while(start < end - 1)
{
middle = (start + end) / 2;
if(value > matrix[middle][middle])
start = middle;
else if(value < matrix[middle][middle])
end = middle;
else
return true;
}
int i = 0, j = end;
while(i < j)
{
middle = (i + j) / 2;
if(value > matrix[end][middle])
i = middle + 1;
else if(value < matrix[end][middle])
j = middle - 1;
else
return true;
}
i = 0, j = end;
while(i < j)
{
middle = (i + j) / 2;
if(value > matrix[middle][end])
i = middle + 1;
else if(value < matrix[middle][end])
j = middle - 1;
else
return true;
}
i = start, j = n-1;
while(i < j)
{
middle = (i + j) / 2;
if(value > matrix[start][middle])
i = middle + 1;
else if(value < matrix[start][middle])
j = middle - 1;
else
return true;
}
i = start, j = n-1;
while(i < j)
{
middle = (i + j) / 2;
if(value > matrix[start][middle])
i = middle + 1;
else if(value < matrix[start][middle])
j = middle - 1;
else
return true;
}
return false;
}
int main(int argc, char *argv[])
{
int key = 4;
cout<<binarySearch(4,key);
return EXIT_SUCCESS;
}