OJ:查找
描述
输入数组长度 n 输入数组 a[1…n] 输入查找个数m 输入查找数字b[1…m] 输出 YES or NO 查找有则YES 否则NO 。
输入描述:
输入有多组数据。 每组输入n,然后输入n个整数,再输入m,然后再输入m个整数(1<=m,n<=100)。
输出描述:
如果在n个数组中输出YES否则输出NO。
示例1
输入:
5
1 5 2 4 3
3
2 5 6
输出:
YES
YES
NO
本题采用二分查找的方式对目标元素进行查找,二分查找部分函数如下:
bool binary_search(int target,int length,int arr[]){
int start=0,end=length-1;
while(start<=end){
int middle=(start+end)/2;
if(arr[middle]<target){
start=middle+1;
}else if(arr[middle]>target){
end=middle-1;
}else{
return true;
}
}
return false;
}
要注意,二分查找仅适用于有序的数组,在使用二分查找前必须要对数组进行排序,对于上面实现的二分查找函数,需要数组为升序的有序数组。
总体代码如下:
#include<iostream>
#include<algorithm>
using namespace std;
bool binary_search(int target,int length,int arr[]){
int start=0,end=length-1;
while(start<=end){
int middle=(start+end)/2;
if(arr[middle]<target){
start=middle+1;
}else if(arr[middle]>target){
end=middle-1;
}else{
return true;
}
}
return false;
}
int main(){
int n;
while(cin>>n){
int arr[n];
for(int i=0;i<n;i++){
cin>>arr[i];
}
sort(arr,arr+n); //排序
int m;
cin>>m;
while(m--){
int target;
cin>>target;
if(binary_search(target,n,arr)){
printf("YES\n");
}else{
printf("NO\n");
}
}
}
return 0;
}
另外,还可以使用vector数组中的 find 函数,具体可参考这篇文章:
vector<>-【王道机试】-查找-查找-北邮