STL <set>练习

找球号(一）

3000 ms  |  内存限制： 65535

3

第一行有两个整数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

代码：

耗时：800ms set的二叉平衡查找树还蛮快！！！

#include <iostream>
#include <set>
#include <cstdio>
using namespace std;

int main()
{
   int x,n,m;
   set<int> s;
   cin>>n>>m;
   for(int i=1;i<=n;i++)
   {scanf("%d",&x); s.insert(x);}
    for(int i=1;i<=m;i++)
    {
       scanf("%d",&x);
      if(s.find(x)!=s.end())
       printf("YES\n");
      else
       printf("YES\n");
    }
    return 0;
}

#include <stdio.h>
#define MAXN 3125010 
int vis[MAXN] = {0} ;
int main()
{
  int m , n , x ; 
  int i ;
  scanf("%d%d", &m , &n ) ;
  for( i = 0 ; i < m ; ++i )
  {
    scanf("%d", &x ) ;
    vis[ x / 32 ] |= 1 << x % 32 ;
  }
  for( i = 0 ; i < n ; ++i )
  {
    scanf("%d", &x ) ;
    if( vis[ x / 32 ] & ( 1 << x % 32 ) )
      printf("YES\n");
    else
      printf("NO\n");
  }
  return 0 ;
}