字典树模板

字典树(trie树)是一种根据前缀在log(n)左右时间内完成查询的数据结构。

可以添加辅助数组来完成特定功能。

插入：

int trie[MAXN][50],sum[MAXN];//sum记录单词出现的次数
char buf[MAXN][50];
int tot;//给节点编号
void insert(char str[])
{
    int root=0;
    for(int i=0;str[i];i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) trie[root][id]=++tot;//如果没有这个前缀，就扩展一个节点，给这个节点编号
        root=trie[root][id];//移到下一个节点
    }
    sum[root]++;
}

查询：

int query(char str[])
{
    int root=0;
    for(int i=0;str[i];i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }
    return sum[root];
}

例题：HDU - 1247 

划分字符串。

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define MAXN 50005
int trie[MAXN][50],sum[MAXN];//sum记录单词出现的次数
char buf[MAXN][50];
int tot;//给节点编号
void insert(char str[])
{
    int root=0;
    for(int i=0;str[i];i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) trie[root][id]=++tot;//如果没有这个前缀，就扩展一个节点，给这个节点编号
        root=trie[root][id];//移到下一个节点
    }
    sum[root]++;
}
int query(char str[])
{
    int root=0;
    for(int i=0;str[i];i++)
    {
        int id=str[i]-'a';
        if(!trie[root][id]) return 0;
        root=trie[root][id];
    }
    return sum[root];
}
int main()
{
    char str[50];
    tot=0;
    memset(trie,0,sizeof trie);
    memset(sum,0,sizeof sum);
    int cnt=0;
    while(~scanf("%s",buf[cnt]))
        insert(buf[cnt++]);
    char c1[100],c2[100];
    for(int i=0;i<cnt;i++)
    {
        int len=strlen(buf[i]);
        for(int j=0;j<len-1;j++)
        {
            memcpy(c1,buf[i],sizeof c1);
            memcpy(c2,buf[i]+j+1,sizeof c2);
            c1[j+1]=0;
            if(query(c1) && query(c2))
            {
                puts(buf[i]);
                break;
            }
        }
    }
    return 0;
}