3道入门字典树例题，以及模板【HDU1251/HDU1305/HDU1671】

HDU1251：http://acm.hdu.edu.cn/showproblem.php?pid=1251

题目大意：求得以该字符串为前缀的数目，注意输入格式就行了。

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

char str[20];
int trie[100000 * 4][30], cnt;
int sum[100000 * 4];

void insert()
{
    int len = strlen(str);
    int rt = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - 'a';
        if(!trie[rt][id])
            trie[rt][id] = ++ cnt;
        rt = trie[rt][id];
        sum[rt] ++;
    }
}

int serach()
{
    int len = strlen(str);
    int rt = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - 'a';
        if(!trie[rt][id])
            return 0;
        rt = trie[rt][id];
    }
    return sum[rt];
}

int main()
{
    while(gets(str))
    {
        if(strlen(str) == 0)
            break;
        insert();
    }
    while(scanf("%s", str) != EOF)
    {
        printf("%d
", serach());
    }
    return 0;
}

HDU1305：http://acm.hdu.edu.cn/showproblem.php?pid=1305

题目大意：字符串之间互相不为前缀，则输出YES 否则NO

#include<stdio.h>
#include<string.h>
#define mem(a, b) memset(a, b, sizeof(a))

int trie[40000][3], tot, k_rt[40000], cnt, sum[40000];
char str[15];

void insert()
{
    int len = strlen(str);
    int root = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - '0';
        if(!trie[root][id])
            trie[root][id] = ++ tot;
        root = trie[root][id];
        sum[root] ++;
    }
    k_rt[++ cnt] = root; //记录每个01字符串结束的位置 
}

void init()
{
    mem(trie, 0);
    mem(k_rt, 0);
    mem(sum, 0);
    cnt = 0, tot = 0;
}

int main()
{
    int k = 1, flag;
    while(scanf("%s", str) != EOF)
    {
        if(str[0] != '9')
            insert();
        else
        {
            flag = 0;
            for(int i = 1; i <= cnt; i ++)
            {
                if(sum[k_rt[i]] >= 2)
                {
                    flag = 1;
                    break;
                }
            }
            if(flag)
                printf("Set %d is not immediately decodable
", k ++);
            else
                printf("Set %d is immediately decodable
", k ++);
            init();
        }
    }
    return 0;
}

HDU1671：http://acm.hdu.edu.cn/showproblem.php?pid=1671

题目大意：与HDU1251是一样的

#include<stdio.h>
#include<string.h>
#define mem(a, b) memset(a, b, sizeof(a))

char str[15];
int trie[100010][12], tot, cnt, k_rt[100010], sum[100010];

void insert()
{
    int len = strlen(str);
    int root = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - '0';
        if(!trie[root][id])
            trie[root][id] = ++ tot;
        root = trie[root][id];
        sum[root] ++;
    }
    k_rt[++ cnt] = root; //记录每个字符串的结束位置 ，通过结束位置上的出现次数判断是否是其他字符串的前缀 
}

void init()
{
    mem(trie, 0);
    mem(sum, 0);
    mem(k_rt, 0);
    tot = 0, cnt = 0;
}

int main()
{
    int T, n, flag;
    scanf("%d", &T);
    while(T --)
    {
        init();
        flag = 0;
        scanf("%d", &n);
        getchar();
        for(int i = 1; i <= n; i ++)
        {
            scanf("%s", str);
            insert();
        }
        for(int i = 1; i <= cnt; i ++)
        {
            if(sum[k_rt[i]] >= 2)
            {
                flag = 1;
                break;
            }
        }
        if(flag)
            printf("NO
");
        else
            printf("YES
");
    }
    return 0;
}

 模板

字典树建树模板：

//建trie树 
void insert()
{
    int len = strlen(str);
    int root = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - '0';
        if(!trie[root][id])
            trie[root][id] = ++ tot;
        root = trie[root][id];
        sum[root] ++; //记录数目则需要 
    }
    pos[++ cnt] = root; //记录每个字符串的结束位置 ，通
    //                    过结束位置上的出现次数判断是否是其他字符串的前缀 
}   //                     若根据输入判断，则不需要这句话，查找输入的字符串最后一个位置的出现次数即可 

字典树查找模板：

//trie树查找 
int serach()//bool serach() , 该字符串为前缀的数目/是否存在以该字符串为前缀的字符串 
{
    int len = strlen(str);
    int rt = 0;
    for(int i = 0; i < len; i ++)
    {
        int id = str[i] - 'a';
        if(!trie[rt][id])
            return 0;//return false;
        rt = trie[rt][id];
    }//达到字符串最后一个字符所在位置 
    return sum[rt]; // return true;
}