南陽OJ-138 找球號（二）(hash表應用)

找球號(二）

時間限制： 1000 ms  |  內存限制：65535 KB

難度： 5

 

描述

在某一國度裏流行着一種遊戲。遊戲規則爲：現有一堆球中，每一個球上都有一個整數編號i(0<=i<=100000000),編號可重複，還有一個空箱子，如今有兩種動做：一種是"ADD"，表示向空箱子裏放m(0<m<=100)個球，另外一種是"QUERY」，表示說出M(0<M<=100)個隨機整數ki(0<=ki<=100000100),分別判斷編號爲ki 的球是否在這個空箱子中（存在爲"YES",不然爲"NO"），先答出者爲勝。如今有一我的想玩玩這個遊戲，但他又很懶。他但願你能幫助他取得勝利。

 

輸入

第一行有一個整數n(0<n<=10000);
隨後有n行;
每行可能出現以下的任意一種形式：
第一種：
一個字符串"ADD"，接着是一個整數m，隨後有m個i；
第二種：
一個字符串"QUERY」，接着是一個整數M，隨後有M個ki；

輸出

輸出每次詢問的結果"YES"或"NO".

樣例輸入

2
ADD 5 34 343 54 6 2
QUERY 4 34 54 33 66

樣例輸出

YES
YES
NO
NO

 1 #include<stdio.h>
 2 #include<string.h>
 3 #define base 1000000    
 4 struct node {
 5     int data, next;
 6 }hash[base+10];
 7 int head[base+10], ip;
 8 void add(int x) {    //添加元素進hash表 
 9     int y = x%base;
10     hash[ip].data = x;
11     hash[ip].next = head[y];
12     head[y] = ip;
13     ip++;
14 }
15 bool query(int x) {//查找數值是否在hash表中 
16     int y = x%base;
17     for (int i = head[y]; i != -1; i = hash[i].next)
18         if (hash[i].data == x)
19             return true;
20     return false;
21 }
22 int main() {
23     int t;
24     memset(head, -1, sizeof(head));
25     scanf("%d", &t);
26     while (t--) {
27         char str[10];
28         int n, m;
29         scanf("%s %d", str, &n);
30         if (str[0] == 'A')
31             while (n--) {
32                 scanf("%d",&m);
33                 add(m);
34             }
35         else
36             while (n--) {
37                 scanf("%d", &m);
38                 if (query(m))
39                     printf("YES\n");
40                 else
41                     printf("NO\n");
42             }
43     }
44 }

 也能夠用stl中的vector ac

 1 #include<iostream>
 2 #include<vector>
 3 #include<cstdio>
 4 using namespace std;
 5 int main() {
 6     vector<bool> v(100000005);
 7     int t;
 8     scanf("%d", &t);
 9     while (t--) {
10         char str[10];
11         int n, m;
12         scanf("%s %d", str, &n);
13         if (str[0] == 'A')
14             for (int i = 0; i < n; i++) {
15                 scanf("%d", &m);
16                 v[m] =  true;
17             }
18         else {
19             for (int i = 0; i < n; i++) {
20                 scanf("%d", &m);
21                 if (v[m])
22                     printf("YES\n");
23                 else
24                     printf("NO\n");
25             }
26                 
27         }
28     }
29 }