2021年2月19日 星期五

d732: 二分搜尋法

 程式碼

#include <iostream>
using namespace std;
int search(int arr[], int len, int x)
{
	int left = 0, right = len - 1, mid = (left + right) / 2;
	while (1)
	{
		if (arr[mid] == x) return mid + 1;
		if (mid < 0 || mid >= len || left >= right) return 0;
		if (arr[mid] > x) right = mid - 1;
		else left = mid + 1;
		mid = (right + left) / 2;
	}
}
int main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(0);
	int n, k, x;
	cin >> n >> k;
	int* arr = new int[n];
	for (int i = 0; i < n; i++)
		cin >> arr[i];
	for (int i = 0; i < k; i++)
	{
		cin >> x;
		cout << search(arr, n, x) << endl;
	}
	return 0;
}

沒有留言:

張貼留言