【算法笔记】问题 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<stdio.h>
#include<algorithm>
using namespace std;
int find_index(int a[],int left,int right,int x);
int main()
{	
	int n,m;
	int a[105];
	int b[105];
	while(scanf("%d",&n)!=EOF)
	{
		for(int i=0;i<n;i++)
		{
			scanf("%d",&a[i]);
		}
		sort(a,a+n);
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			scanf("%d",&b[i]);
		}
		
		for(int i=0;i<m;i++)
		{
			if(find_index(a,0,n-1,b[i])!=-1)
			{
				printf("YES\n");
			}
			else
			{
				printf("NO\n");
			}
		}
	}
	return 0;
}
int find_index(int a[],int left,int right,int x)
{
	int mid;
	while(left<=right)
	{
		mid=(left+right)/2;
		if(a[mid]==x) return mid;
		if(a[mid]<x) left=mid+1;
		else right=mid-1;
	}
	return -1;
}

tips：
没有说明输入的序列是有序的。