nyoj--290--动物统计加强版

最新推荐文章于 2018-10-11 17:38:51 发布

原创最新推荐文章于 2018-10-11 17:38:51 发布 · 405 阅读

0 ·

CC 4.0 BY-SA版权

算法_字典树专栏收录该内容

3 篇文章

订阅专栏

本文介绍了一种使用字典树高效解决大规模动物名称统计问题的方法。通过构建字典树并记录每个节点的出现次数，可以快速找出出现频率最高的动物名称。

动物统计加强版

时间限制：3000 ms | 内存限制：150000 KB

难度：4

描述

在美丽大兴安岭原始森林中存在数量繁多的物种，在勘察员带来的各种动物资料中有未统计数量的原始动物的名单。科学家想判断这片森林中哪种动物的数量最多，但是由于数据太过庞大，科学家终于忍受不了，想请聪明如你的ACMer来帮忙。

输入

第一行输入动物名字的数量N(1<= N <= 4000000),接下来的N行输入N个字符串表示动物的名字（字符串的长度不超过10,字符串全为小写字母，并且只有一组测试数据）。

输出

输出这些动物中最多的动物的名字与数量，并用空格隔开（数据保证最多的动物不会出现两种以上）。

样例输入

10
boar
pig
sheep
gazelle
sheep
sheep
alpaca
alpaca
marmot
mole

样例输出

sheep 3

解题思路：

开始用map做了一遍，果断超时。后来改用字典树做，ac了，思路也很简单，就是在声明Trie节点的时候，加入一个num成员，用来记录各个单词出现的个数，在插入单词

的同时，进行计数，另外声明一个int 型的ans，记录出现的单词的最大次数，char类型的数组 b，记录出现次数最多的单词，当单词全部插入的时候，结果也就出来了，这里有两种字典树实现的方法，一种是利用树组实现，另一种是用链表实现，可以根据自己的情况进行选择；

链表代码：

 C++ Code 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

#include<bits/stdc++.h>
#define MAX 26
using namespace std;
typedef struct TrieNode           ///Trie节点声明
{
    int  num;                   ///记录这个单词的个数
    struct TrieNode *next[MAX];   ///儿子分支
} Trie;
int ans = -1;
char b[15];
void insert1(Trie *root, char *s)
{
    char w[15];
    strcpy(w, s);
    if(root == NULL || *s == '\0')
        return ;
    Trie *p = root;
    while(*s != 0)
    {
        if(p->next[*s - 'a'] == NULL)
        {
            Trie *temp = (Trie *)malloc(sizeof(Trie));
            for(int i = 0; i < MAX; i++)
                temp->next[i] = NULL;
            temp->num = 0;
            p->next[*s - 'a'] = temp;
            p = p->next[*s - 'a'];
        }
        else
            p = p->next[*s - 'a'];
        s++;
    }
    p->num++;
    if(p->num > ans)
    {
        ans = p->num;
        strcpy(b, w);
    }
}
int main()
{
    int n;
    char a[15];
    Trie *root = (Trie *)malloc(sizeof(Trie));
    for(int i = 0; i < MAX; i++)
        root->next[i] = NULL;
    scanf("%d", &n);
    for(int i = 0; i < n; i++)
    {
        scanf("%s", a);
        insert1(root, a);
    }
    printf("%s %d\n", b, ans);
    return 0;
}

数组代码：

 C++ Code 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

#include <bits/stdc++.h>
using namespace std;
#define idx(x) x - 'a';
const int MAXN = 1e6;
struct Trie
{
    int next[26];///存储下一个节点的情况
    int val;///数据域
    char w[15];
} tree[MAXN];

int maxn = -1;
char ans[15];
int nxt, T;///nxt表示节点的编号
char str[15];
int add()
{
    ///开辟新的节点，并进行初始化
    memset(&tree[nxt], 0, sizeof(Trie));
    return nxt++;
}
void Insert(char *s)
{
    int rt = 0, len = strlen(s); ///rt=0，每次插入都从根节点开始查询，
    for(int i = 0; i < len; i++)
    {
        int c = idx(s[i]);///把字母转化为数字
        if(!tree[rt].next[c])///如果这个节点还不存在，就用add函数开辟新的节点
            tree[rt].next[c] = add();
        rt = tree[rt].next[c];
    }
    tree[rt].val++;
    strcpy(tree[rt].w, s);
    if(tree[rt].val > maxn)
    {
        maxn = tree[rt].val;
        strcpy(ans, tree[rt].w);
    }
}
int main()
{
    memset(&tree[0], 0, sizeof(Trie));
    nxt = 1;
    scanf("%d", &T);
    maxn = -1;
    ans[15];
    while(T--)
    {

        scanf("%s", str);
        Insert(str);
    }
    printf("%s %d\n", ans, maxn);
    return 0;
}