hdu 1251 统计难题

Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 131070/65535 K (Java/Others)
Total Submission(s): 57491 Accepted Submission(s): 20104

Problem Description

Ignatius最近遇到一个难题,老师交给他很多单词(只有小写字母组成,不会有重复的单词出现),现在老师要他统计出以某个字符串为前缀的单词数量(单词本身也是自己的前缀).

Input

输入数据的第一部分是一张单词表,每行一个单词,单词的长度不超过10,它们代表的是老师交给Ignatius统计的单词,一个空行代表单词表的结束.第二部分是一连串的提问,每行一个提问,每个提问都是一个字符串.

注意:本题只有一组测试数据,处理到文件结束.

Output

对于每个提问,给出以该字符串为前缀的单词的数量.

Sample Input

banana band bee absolute acm ba b band abc

Sample Output

2 3 1 0

Author

Ignatius.L

Recommend

Ignatius.L | We have carefully selected several similar problems for you: 1075 1247 1671 1298 1800

查前缀相同的单词个数，典型的字典树题目，字典树有两种实现方法，链表实现，每个结点最多有26个儿子，但是一旦结点存在，必然有二十六个空指针被分配内存。链表实现很浪费内存，效率也低，所以本地链表实现会内存溢出。

链表实现代码：

#include <iostream>
#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
typedef struct trie Trie;
struct trie {
    int Pre;
    Trie *Next[26];
}*head;
inline Trie* build() {
    Trie *p = (Trie *)malloc(sizeof(Trie));
    p -> Pre = 0;
    for(int i = 0;i < 26;i ++) {
        p -> Next[i] = NULL;
    }
    return p;
}
inline void Insert(char *s) {
    Trie* p = head;
    int i = 0;
    while(s[i]) {
        int d = s[i] - 'a';
        if(p -> Next[d] == NULL) {
            p -> Next[d] = build();
        }
        p = p -> Next[d];
        p -> Pre ++;
        i ++;
    }
}
int Pre_Query(char *s) {
    Trie* p = head;
    int i = 0;
    while(s[i]) {
        int d = s[i] - 'a';
        if(p -> Next[d] == NULL) {
            return 0;
        }
        p = p -> Next[d];
        i ++;
    }
    return p -> Pre;
}
int main() {
    head = build();
    char s[12];
    while(gets(s) && s[0]) {
        Insert(s);
    }
    while(~scanf("%s",s)) {
        printf("%d
",Pre_Query(s));
    }
}

下面说一下数组实现，按照字母出现时间，给定每个字母一个数字作为他的位置i，然后根据他是父节点的第几个儿子，再给定一个数字j，如此来确定这个字母。后面的j也可以根据字母表的顺序，皆可。这样根节点的位置其实就是0，他不表示任何字母，他的儿子表示各自开头的单词，用一个二维数组来记录，前缀个数用一个对应出现时间的一维数组来记录。

数组实现代码：

#include <iostream>
#include <cstdio>
#include <map>
#include <algorithm>
using namespace std;
int trie[400001][26],pre[400001],pos;
inline void Insert(char *s) {
    int i = 0,c = 0;
    while(s[i]) {
        int d = s[i] - 'a';
        if(!trie[c][d]) {
            trie[c][d] = ++ pos;
        }
        c = trie[c][d];
        pre[c] ++;
        i ++;
    }
}
inline int Pre_Query(char *s) {
    int i = 0,c = 0;
    while(s[i]) {
        int d = s[i] - 'a';
        if(!trie[c][d]) {
            return 0;
        }
        c = trie[c][d];
        i ++;
    }
    return pre[c];
}
int main() {
    char s[12];
    while(gets(s) && s[0]) {
        Insert(s);
    }
    while(~scanf("%s",s)) {
        printf("%d
",Pre_Query(s));
    }
}