HDOJ 1251 统计难题 (trie）

最新推荐文章于 2019-08-09 20:14:37 发布

Cifer

最新推荐文章于 2019-08-09 20:14:37 发布

阅读量458

点赞数

分类专栏：树 ACM做题笔记文章标签： null struct

本文链接：https://blog.csdn.net/ulquiorra0cifer/article/details/7756835

版权

ACM做题笔记同时被 2 个专栏收录

120 篇文章 0 订阅

订阅专栏

树

2 篇文章 0 订阅

订阅专栏

题目链接：(—_—) zZ

字典树

code：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
typedef struct node
{
    struct node *next[27];
    int count;
}trie;
trie *root = NULL;
char str[12];
int len = 0;
void create_trie()
{
    int i = 0, j = 0, id = 0;
    trie *p = root, *q = NULL;
    for(i = 0; i<len; i++)
    {
        id = str[i]-'a';
        if(p->next[id] == NULL)
        {
            q = (trie *)malloc(sizeof(trie));
            q->count = 1;
            for(j = 0; j<26; j++)
                q->next[j] = NULL;
            p->next[id] = q;
            p = p->next[id];
        }
        else
        {
            p->next[id]->count++;
            p = p->next[id];
        }
    }
}
int find_trie()
{
    int i = 0, id = 0;
    trie *p = root;
    for(i = 0; i<len; i++)
    {
        id = str[i]-'a';
        p = p->next[id];
        if(p == NULL)
            return 0;
    }
    return p->count;
}
int main()
{
    int i = 0;
    root = (trie *)malloc(sizeof(trie));
    for(i = 0; i<26; i++)
        root->next[i] = NULL;
    while(gets(str))
    {
        len = strlen(str);
        if(!len)
            break;
        create_trie();
    }
    while(gets(str))
    {
        len = strlen(str);
        printf("%d\n", find_trie());
    }
    return 0;
}