字典树模板

最新推荐文章于 2019-11-19 17:00:04 发布

weixin_30527323

最新推荐文章于 2019-11-19 17:00:04 发布

阅读量52

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/sola1994/p/4731861.html

struct node
{
    int next[27];
    int v;
    void init()
    {
        v=0;
        memset(next,-1,sizeof(next));
    }
};

node L[1000500];
int tot=0;

void add(char s[],int len)
{
    int now=0;
    for(int i=0;i<len;i++)
    {
        int temp=s[i]-'s';
        int next=L[now].next[temp];
        if(next==-1)
        {
            next=++tot;
            L[next].v=-1;
            L[next].init();
            L[now].next[temp]=next;
        }
        now=next;
    }
    L[now].v=0;
}

int query(char s[],int len)
{
    int now=0;
    for(int i=0;i<len;i++)
    {
        int temp=s[i]-'a';
        int next=L[now].next[temp];
        if(next==-1) return -1;
    }
    return L[now].v;
}

转载于:https://www.cnblogs.com/sola1994/p/4731861.html