1. 程式人生 > >字典樹 trie

字典樹 trie

具體原理:劉汝佳《演算法競賽入門經典訓練指南》P208

作用:字典樹又叫單詞查詢樹(Trie)或字首樹,是一個實現字串。你可以在O(m)(m為所給單詞的長度)時間內判斷出該單詞是否屬於字典

程式碼模板:
 

struct Trie
{
    int ch[maxnode][sigma_size];//ch[i][j]==k表示第i個節點的第j個兒子是節點k
    int val[maxnode];//val[i]==x表示第i個節點的權值為x
    int sz;//字典樹一共有sz個節點,從0到sz-1標號
 
    //初始化
    void clear()
    {
        sz=1;
        memset(ch[0],0,sizeof(ch[0]));//ch值為0表示沒有兒子
    }
 
    //在字典樹中插入單詞s,但是如果已經存在s單詞會重複插入且覆蓋權值
    //所以insert前需要判斷一下是否已經存在s單詞了
    void insert(string s)
    { 
		int u=0,n=s.length();
        for(int i=0;i<n;i++)///建立字典樹
        {
            int id=s[i]-'a';
            if(ch[u][id]==0)//無該兒子
            {
                ch[u][id]=sz;
                memset(ch[sz],0,sizeof(ch[sz]));
                val[sz++]=0;
            }
            u=ch[u][id];
        }
        val[u]=n;
    }
 
    //在字典樹中查詢單詞s
    bool find(string s)
    {
        int n=s.length(),u=0;
        for(int i=0;i<n;i++)
        {
            int id=s[i]-'a';
            if(ch[u][id]==0)
                return false;
            u=ch[u][id];
     
        }
        return val[u];
    }
};
Trie trie;

經典例題: 

 

HDU 1251 統計難題                    字典樹簡單應用(找出所有以字串s為字首的單詞個數)

Uva1401 Remember the Word    字典樹簡單應用(求一個字串能有多少種方式通過字典中的單詞構成)

HDU 1671 Phone List                   字典樹 (多種做法:問你字典中是否有一個字串是其他字串的字首)

HDU 1247 Hat’s Words(              字典樹Trie):單詞匹配

異或(兩題一樣,推薦第二道)

CH 1602 The XOR Largest Pair 字典樹+異或

NBUT-1597  Find MaxXorSum     經典字典樹求異或最大值(資料量大)

 

 

POJ 3764 The xor-longest Path    dfs+字典樹求最大異或