【字符串匹配】字典树模板

字典树

 

tree.insert(s,v); 插入字符串s，其中v为字符串的序号（从1开始）。

tree.exist(s); 检测字符串s是否存在。

 

struct Trie
{
    int ch[1000005][26];
    int val[1000005];
    int size;
    Trie()
    {
        size=1;
        memset(ch[0],0,sizeof(ch[0]));
    }
     
    int idx(char c)
    {
        return c-'a';
    }
     
    void insert(char *s,int v)
    {
        int u=0,len=strlen(s);
        for (int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if (!ch[u][c])
            {
                memset(ch[size],0,sizeof(ch[size]));
                val[size]=0;
                ch[u][c]=size++;
            }
            u=ch[u][c];
        }
        val[u]=v;
    }
    
    bool exist(char *s)
    {
        int u=0,len=strlen(s);
        for (int i=0;i<len;i++)
        {
            int c=idx(s[i]);
            if (!ch[u][c])
            {
                return false;
            }
            u=ch[u][c];
        }
        if (val[u]) return true;
        else return false;
    }
} tree;