1000: 恶意IP 课程作业

1000: 恶意IP

Time Limit: 1 Sec  Memory Limit: 16 MB

Description

Water同学最近好不容易学会了用Tornado建起一个个人的Website，并且成功上线了。

来访用户逐渐增多，但Water发现总有些恶意用户很喜欢刷屏，总是回复些评论如“楼主不要放弃治疗！”，“楼主药不能停！”之类的。Water感受到了这个世界满满的恶意，他很不爽，决定将这部分恶意用户过滤掉。

他已经掌握到这些用户的IP了，但是过滤IP这件事情对于数据结构挂了的他来说实在是有些困难，所以他来找你帮忙了！

IP格式为 a.b.c.d , 其中 a,b,c,d均为[0,255]之间的整数。

Input

只有一组数据。第一行为一个整数N ［0, 1 000 000］，代表恶意IP列表的长度。接下来N行是N个恶意IP。
然后紧随一个整数M [0, 1 000 000]，代表访问IP的长度。接下来M行是M个来访IP。
你需要判断该来访IP是否在恶意IP列表中。

Output

如果来访IP是恶意IP，则输出 FILTER，否则输出 PASS。

Sample Input

5
233.233.233.233
250.250.250.250
10.20.30.40
123.255.123.255
172.18.182.69
6
10.123.128.245
233.233.233.233
172.18.182.253
102.30.40.50
172.18.182.96
172.18.182.69

Sample Output

PASS
FILTER
PASS
PASS
PASS
FILTER

这是我们数据结构的一道题，这道题本来不难，只要时间内存足够一个map就可以了，关键是时间要求1s内存是16MB，并且坑的是，测试数据量1 000 000组。那么这就要讲究策略了。首先不能用c++的STL和输入输出流还有string吧。。这些都是很耗空间和时间的。要效率就得选择C语言了。
然后是搜索问题，我们要根据字符串来搜索。在这里不管是快排还是归并都是达不到要求的，因为数据量太大。那么要选择什么呢？肯定哈希啦。。
所以下面就是怎么构造哈希表的问题了。因为键值是字符串不能当数组的下标，而且建数组的话也要注意大小问题。我选择的是BKDRHash函数，因为这个函数还是比较常用的。函数如下：

unsigned int BKDRHash(char *str)
{
    unsigned int seed = 131; // 31 131 1313 13131 131313 etc..
    unsigned int hash = 0;

    while (*str)
    {
        hash = hash * seed + (*str++);
    }

    return (hash & 0x7FFFFFFF);
}

这里因为数据量最大是 1 000 000， 那么我们就构建一个 1 000 000 大小的数组吧。然后因为当字符串长的时候 BKDRHash()函数的返回值其实是很大的，所以返回值要对 1 000 000取余。

还有一个要注意的问题是冲突问题，一般情况下我一般用拉链法。。所以就拉链了。。。

下面是我的代码实现，因为写得比较急有点挫。。。

#include <iostream>
#include <stdio.h>
#include <string.h>

using namespace std;

unsigned int BKDRHash(char *str) {
    unsigned int seed = 131;
    unsigned int hash = 0;

    while (*str) {
        hash = hash * seed + (*str++);
    }

    return (hash & 0x7FFFFFFF);
}

struct Node {
    char* value;
    Node* next;
    //value is link to a list which has conflict elements
    Node(char* value_ = NULL){
        value = value_;
        next = NULL;
    }
};

Node **arr = new Node*[1000001];

void initializeHash(int N) {
    //we can't only use a char[] here because the
    //address of str will not change in this way, 
    //and the elements will be the same because
    //they are just the same element
    char* str;
    for (int i = 0; i != N; i++) {
        str = new char[17];
        scanf("%s", str);
        unsigned int temp = BKDRHash(str);
        temp %= 1000000;

        if (arr[temp] == NULL) {
            arr[temp] = new Node(str);
        } else {
            Node* root = arr[temp];
            while (root->next != NULL) {
                root = root->next;
            }
            root->next = new Node(str);
        }
    }
}

void judgeIp(char* str) {
    unsigned int temp = BKDRHash(str);
        temp %= 1000000;
        if (arr[temp] == NULL) {
            printf("PASS\n");
        } else {
            bool flag = true;
            Node* root = arr[temp];
            while (root != NULL) {
                if (strcmp((root->value), str) == 0) {
                    flag = false;
                    printf("FILTER\n");
                    break;
                }    
                root = root->next;
            }
            if (flag)
               printf("PASS\n");
        }
}
int main(int argc, char const *argv[])
{
    int N;
    scanf("%d", &N);
    for (int i = 0; i != 1000001; i++)
        arr[i] = NULL;
    initializeHash(N);

    char str[17];
    scanf("%d", &N);
    for (int i = 0; i != N; i++) {        
        scanf("%s", str);
        judgeIp(str);
    }
    return 0;
}