找球号(一）
时间限制：3000 ms | 内存限制：65535 KB
难度：3
- 描述
- 在某一国度里流行着一种游戏。游戏规则为：在一堆球中，每个球上都有一个整数编号i(0<=i<=100000000),编号可重复，现在说一个随机整数k(0<=k<=100000100)，判断编号为k的球是否在这堆球中（存在为"YES",否则为"NO"），先答出者为胜。现在有一个人想玩玩这个游戏，但他又很懒。他希望你能帮助他取得胜利。
- 输入
- 第一行有两个整数m，n(0<=n<=100000，0<=m<=1000000)；m表示这堆球里有m个球，n表示这个游戏进行n次。
接下来输入m+n个整数，前m个分别表示这m个球的编号i，后n个分别表示每次游戏中的随机整数k - 输出
- 输出"YES"或"NO"
- 样例输入
-
6 4
23 34 46 768 343 343
2 4 23 343 - 样例输出
-
NO
NO
YES
YES
这题两种方法，一种简单但速度较慢，一种速度很快，但是要自己维护一棵树
第一种使用stl的set，用时1700MS，代码只有30行，简单易用
#include<iostream>
#include<set>
#include<string>
using namespace std;
int main()
{
set<int> set_num;
int n , m;
int num;
cin >> n;
cin >> m;
while(n--)
{
cin >> num;
set_num.insert (num);
}
while(m--)
{
cin>>num;
if(set_num.count (num))
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
return 0;
}
第二种方法是自己维护一棵二分查找树，代码写了100多行，比较罗嗦，但用时300ms，差距挺大的
#include<stdio.h>
#include<stdlib.h>
typedef struct NODE
{
int num;
bool flag;
struct NODE * left , *right;
}Node;
Node haxi[5009];
void insert(Node * head , Node * new_node)
{
if(head->flag == 0)
{
head->num = new_node->num;
head->flag = 1;
free(new_node);
}
else
{
for( ; head != NULL; )
{
if(new_node->num > head->num)
{
if(head->right == NULL)
{
head->right = new_node;
return ;
}
else
head = head->right;
}
else if(new_node->num < head->num)
{
if(head->left == NULL)
{
head->left = new_node;
return ;
}
else
head = head->left;
}
else
{
free(new_node);
return ;
}
}
}
}
int search_node(Node * head , int num)
{
if(head->flag == 0)
return 0;
else
{
for( ; head != NULL; )
{
if(num > head->num)
{
if(head->right == NULL)
return 0;
else
head = head->right;
}
else if(num < head->num)
{
if(head->left == NULL)
return 0;
else
head = head->left;
}
else
return 1;
}
}
}
int main()
{
int i , j , num;
Node * new_node; scanf("%d %d" , &i , &j);
while(i--)
{
new_node = (Node *) calloc(1 , sizeof(Node));
scanf("%d" , &new_node->num );
insert(&haxi[new_node->num % 5009] , new_node);
}
while(j--)
{
scanf("%d" , &num);
if(search_node(&haxi[num % 5009] , num) == 1)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}