搜索(三)
总提交:1472 测试通过:307
描述
给定整数序列,查询多个整数是否存在。
输入
输入多行:
l 第1行给出整数序列大小n,查询次数m(待查询的整数可以重复), 1≤n≤100000,1≤m≤100000
l 第2行给出整数序列,每个整数绝对值不超1000000
l m行,每1行给出指定整数
输出
对于每个测试用例:
l 输出m行,每1行给出一个查询结果,指定整数存在则输出“Yes”,否则输出“No”
注意:输出部分的结尾要求包含一个多余的空行。
样例输入
2 2
1 2
2
3
样例输出
Yes
No
题目来源
算法与数据结构设计考核赛2009
代码:
#include <iostream>
using namespace std;
int a[1000000];
void Swap(int &a,int &b); //用于数据的交换,用在快排函数中
void QuickSort(int *a,int left,int right); //快排
bool Select(int *a,int a_length,int x); //在a中查询 x 是否存在
int main()
{
int n,m,element;
cin>> n >> m;
for(int i = 0; i < n ; i++)
cin >> a[i];
QuickSort(a,0,n - 1);
while(m-->0)
{
cin >> element;
if(Select(a,n,element))cout << "Yes"<<endl;
else cout << "No"<<endl;
}
return 0;
}
void Swap(int &a,int &b)
{
int tmp = a;
a = b;
b = tmp;
}
void QuickSort(int *a,int left,int right)
{
if(left <= right)
{
int i = left;
int j = right + 1;
do
{
do i++;while(a[left] > a[i]);
do j--;while(a[left] < a[j]);
if(i < j)Swap(a[i],a[j]);
}while(i < j);
Swap(a[j],a[left]);
QuickSort(a,left,j - 1);
QuickSort(a,j + 1,right);
}
}
bool Select(int *a,int a_length,int x)
{
int left = 0;
int right = a_length - 1;
int middle = 0;
while(left <= right)
{
middle = (left + right)/2;
if(a[middle] > x)right = middle - 1;
else if(a[middle] < x)left = middle + 1;
else return true;
}
return false;
}
快排和二分搜索结合比较高效一下,以后常尝试。。。