算法笔记练习 4.5 二分 问题 C: 查找

算法笔记练习 题解合集

题目链接

题目

题目描述
输入数组长度 n
输入数组 a[1…n]
输入查找个数m
输入查找数字b[1…m]
输出 YES or NO 查找有则YES 否则NO 。

输入
输入有多组数据。
每组输入n，然后输入n个整数，再输入m，然后再输入m个整数（1<=m<=n<=100）。

输出
如果在n个数组中输出YES否则输出NO。

样例输入

6
3 2 5 4 7 8
2
3 6

样例输出

YES
NO

代码

#include 
#include 
// qsort 排序函数 
int cmpInt(const void *a, const void *b){
	return *(int*)a > *(int*)b; 
}
// 在长度 n 的数组 a 中找 x，找到返回下标，找不到返回 -1
int binarySearch(int n, int *a, int x){
	int ret = -1;
	int left = 0;
	int right = n - 1;
	int middle;
	while (left <= right){
		middle = left + (right - left) / 2;
		if (a[middle] == x){
			ret = middle;
			break;
		} else if (a[middle] > x)
			right = middle - 1;
		else
			left = middle + 1;
	}
	return ret;
} 

int main(){
	int n, m, i;
	while (scanf("%d", &n) != EOF){
		int nums[n];
		for (i = 0; i < n; ++i)
			scanf("%d", &nums[i]);
		qsort(nums, n, sizeof(int), cmpInt);
		
		int input;
		scanf("%d", &m);
		while (m--){
			scanf("%d", &input);
			if (binarySearch(n, nums, input) != -1)
				printf("YES\n");
			else
				printf("NO\n");
		} 
	} 
	return 0;
}