二分类题目

/*问题1
leetcode 33. 搜索旋转排序数组
*/
int search(vector<int>& nums, int target) {
	int len = nums.size();
	if (len == 0) return -1;

	int left = 0, right = len - 1;
	int mid = 0;// 分割
	// 将数组一分为二，其中一定有一个是有序的，另一个可能是有序，也能是部分有序。此时有序部分用二分法查找。无序部分再一分为二，其中一个一定有序，另一个可能有序，可能无序。就这样循环.

	int i = 0, j = 0;
	int m = 0; // 二分
	while (left <= right) {
		mid = (left + right) / 2;

		if (nums[mid] < nums[left]) {// 左边无序 右边有序
			// mid -> j用二分
			i = mid; j = right;
			while (i <= j) {
				m = (i + j) / 2;
				if (nums[m] == target) {
					return m;
				}
				else if (nums[m] < target) {
					i = m + 1;
				}
				else {
					j = m - 1;
				}
			}
			right = mid - 1;
		}
		else {// 左边有序 右边无序
			i = left; j = mid;
			while (i <= j) {
				m = (i + j) / 2;
				if (nums[m] == target) {
					return m;
				}
				else if (nums[m] < target) {
					i = m + 1;
				}
				else {
					j = m - 1;
				}
			}
			left = mid + 1;
		}
	}
	return -1;

}


/*问题2
已知一个有序矩阵mat，同时给定矩阵的大小n和m以及需要查找的元素x，且矩阵的行和列都是从小到大有序的。
设计查找算法返回所查找元素的二元数组，代表该元素的行号和列号(均从零开始)。保证元素互异。
输入：[[1,2,3],[4,5,6]],2,3,6
返回值：[1,2]
*/
vector<int> findElement(vector<vector<int> > mat, int rows, int cols, int x) {
	// write code here
	vector<int> ret;
	if (rows == 0 || cols == 0) return ret;

	int row = 0;
	int col = -1;
	// 二分查找 x
	// [1, 2, 3]
	// [4, 5, 6]   6  // 行的查找需要定制
	// [7  8  9]
	// [10 11 12]
	int left = 0, right = rows-1;
	
	for (row = 0; row < rows; ++row) {
		left = 0; right = cols - 1;
		while (left <= right) {
			int mid_y = (left + right) / 2;  // 行的中间位置
			if (x > mat[row][mid_y]) {
				left = mid_y + 1;
			}
			else if (x < mat[row][mid_y]) {
				right = mid_y - 1;
			}
			else if (x == mat[row][mid_y]) {
				col = mid_y;
				break;
			}
		}
		if (col != -1) break;
	}

	if (col == -1) return ret;

	ret.emplace_back(row);
	ret.emplace_back(col);
	

	return ret;
}