字典树的指针模板与数组模板

第一个模板便是指针字典树的模板:

#include <iostream>
#include <cstdio>
#include <cstring>
const int MAX=26;
struct Tire{
Tire *next[MAX];
int v;

};
Tire *root;
void createTrie(char *str)
{
    int len = strlen(str);
    Trie *p = root, *q;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'0';
        if(p->next[id] == NULL)
        {
            q = (Trie *)malloc(sizeof(Trie));
            q->v = 1;    //初始v==1
            for(int j=0; j<MAX; ++j)
                q->next[j] = NULL;
            p->next[id] = q;
            p = p->next[id];
        }
        else
        {
            p->next[id]->v++;
            p = p->next[id];
        }
    }
    p->v = -1;   //若为结尾,则将v改成-1表示
}
int findTrie(char *str)
{
    int len = strlen(str);
    Trie *p = root;
    for(int i=0; i<len; ++i)
    {
        int id = str[i]-'0';
        p = p->next[id];
        if(p == NULL)   //若为空集,表示不存以此为前缀的串
            return 0;
        if(p->v == -1)   //字符集中已有串是此串的前缀
            return -1;
    }
    return -1;   //此串是字符集中某串的前缀
}
int dealTrie(Trie* T)
{
    int i;
    if(T==NULL)
        return 0;
    for(i=0;i<MAX;i++)
    {
        if(T->next[i]!=NULL)
            deal(T->next[i]);
    }
    free(T);
    return 0;
}
using namespace std;
int main(){



return 0;
}
第二个模板是数组模板

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node{
    int next[27];
    int v;
    void init(){
        v=0;
        memset(next,-1,sizeof(next));
    }
};
struct node L[1000000];
int tot=0;

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

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



你可能感兴趣的:(字典树的指针模板与数组模板)