题目
又到过年了,狗熊岭的动物们都忙碌起来,张灯结彩准备过年。李老板却要光头强砍掉一些百年美人松回去。美人松都是很高的,但是也不会超过长整型(long long)。现在光头强看到丛林里有N颗美人松按照从矮到高排好了,当然每棵松的高度都是已知的。李老板要问光头强M次:每次询问高度为K的美人松是否存在?
输入
第一行,两个正整数N,M,1<=N<=10^6,1<=M<=10^3;
第二行,N个正整数,之间用一个空格隔开,表示N棵美人松的高度;
第三行M个正整数k,表示M个询问,每次询问高度为K的美人松是否存在。
(提示:要用scanf读入,否则会超时)
输出
一行M个正整数,之间用一个空格隔开,分别表示对应每次询问高度为K的树的查询结果,如果存在,则输出1,不存在则输出0
样例输入
5 2 2 3 3 4 5 3 6
样例输出
1 0
#include <iostream>
using namespace std;
#define MX 1000000
long long int a[MX + 10]; //定义一个数组,用于容纳百年松
int main()
{
long long int n, m, b[1100],mid,L,R,found;
cin >> n >> m;
for (int i = 1; i <= n; i++)
{
scanf("%lld", &a[i]);
}
for (int i = 1; i <= m; i++)
{
scanf("%lld", &b[i]);
}
for(int i=1;i<=m;i++)//m次询问
{
L = 1;
R = n;
found = 0;
while(L<=R)//二分法查找
{
mid = (L+R)/2;//找到区间中间的一个数
if(a[mid]>b[i])
{
R = mid - 1;//舍去右半区间,在左半区间查找
}
else if(a[mid]<b[i])
{
L = mid + 1;//舍去左半区间,在右半区间查找
}
else
{
found = 1;
break;
}
}
printf("%lld%s",found," ");
}
return 0;
}
数据较大时,可采取二分法查找,减少循环次数。