力扣练习——55 搜索二维矩阵 II

55 搜索二维矩阵 II

1.问题描述
编写一个高效的算法来搜索 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。

可使用以下main函数：

int main()

{

vector<vector<int> > matrix;

int target;

int m,n,e;

cin>>m;

cin>>n;

for(int i=0; i<m; i++)

{

    vector<int> aRow;

    for(int j=0; j<n; j++)

    {

        cin>>e;

        aRow.push_back(e);

    }

    matrix.push_back(aRow);

}

cin>>target;

bool res=Solution().searchMatrix(matrix,target);

cout<<(res?"true":"false")<<endl;

return 0;

}

2.输入说明
首先输入matrix的行数m、列数n，

然后输入m行，每行n个整数。

最后输入一个整数target。
3.输出说明
输出true或false
4.范例
输入
5 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
5
输出
true
5.代码

#include <iostream>
#include <queue>
#include <cstdlib>
#include <cstring>
#include<algorithm>

using namespace std;

int binarySearch(vector<int>nums, int target)  //注意，这里不能把int改成bool,否则会报错
{
	int low = 0;
	int high = nums.size() - 1;
	while (low <= high)
	{
		int mid = (low + high) / 2;
		if (nums[mid] < target)
			low = mid + 1;
		else if (nums[mid] > target)
			high = mid - 1;
		else
			return mid;
	}
	return -1;
}

bool searchMatrix(vector<vector<int> >matrix, int target)
{
	//每行进行二分查找
	
	int m = matrix.size();//行数
	int n = matrix[0].size();//列数
	for (int i = 0; i < m; i++)
	{
		if (matrix[i][0] > target)//若该行第一个元素大于target,则该行及后面所有行都不用考虑
			break;
		if (matrix[i][n - 1] < target)//该行最后一个元素值小于target ，则继续搜下行
			continue;
		int col = binarySearch(matrix[i], target);//对第i行搜索target
		if (col != -1)
		{
			//cout << "col=" << col << endl;
          return true;
		}
			
	}
	return false;
}

int main()

{

	vector<vector<int> > matrix;

	int target;

	int m, n, e;

	cin >> m;

	cin >> n;

	for (int i = 0; i < m; i++)

	{

		vector<int> aRow;

		for (int j = 0; j < n; j++)

		{

			cin >> e;

			aRow.push_back(e);

		}

		matrix.push_back(aRow);

	}

	cin >> target;

	bool res = searchMatrix(matrix, target);

	cout << (res ? "true" : "false") << endl;

	return 0;

}