nyoj 86 找球号（一）

点击打开链接

找球号(一）

时间限制：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;
}