字典树模板

字典树  

这一个比较简单  现在回顾也挺简单 

直接给出模板 有什么不会的直接去模板里理解

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

const int maxn = 500005;
int tire[maxn][26];
int flag[maxn];
int tot;

void init() {
    memset(tire, 0, sizeof(tire));
    memset(flag, 0, sizeof(flag));
    tot = 1;
}

void add(char *s, int rt) {
    for(int i = 0; s[i]; i++) {
        int x = s[i] - '0';
        if(tire[rt][x] == 0) {
            tire[rt][x] = tot++;
        }
        rt = tire[rt][x];
        flag[rt]++;//计算以当前字符串做前缀的字符串的数目
    }
}

int cal(char *s, int rt) {
    for(int i = 0; s[i]; i++) {
        int x = s[i] - '0';
        if(tire[rt][x] == 0) {
            return 0;
        }
        rt = tire[rt][x];
    }
    return flag[rt];
}

char s[15];
int main() {
    init();
    while(gets(s)) {
        if(strlen(s) == 0) {
            break;
        }
        add(s, 0);
    }
    while(EOF != scanf("%s",s)) {
        printf("%d
",cal(s, 0));
    }
}

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

struct Node {
    Node *next[26];
    int cnt;
    Node() {
        for(int i = 0; i < 26; i++) {
            next[i] = NULL;
        }
        cnt = 0;
    }
}root;

void add(char *s) {
    Node *p = &root;
    for(int i = 0; s[i]; i++) {
        if(p -> next[s[i] - 'a'] == NULL) {
            p -> next[s[i] - 'a'] = new Node();
        }
        p = p -> next[s[i] - 'a'];
        p -> cnt++;
    }
}

int cal(char *s) {
    Node *p = &root;
    for(int i = 0; s[i]; i++) {
        if(p -> next[s[i] - 'a'] == NULL) {
            return 0;
        }
        p = p -> next[s[i] - 'a'];
    }
    return p -> cnt;
}

void Free(Node *p) {
    for(int i = 0; i < 26; i++) {
        if(p -> next[i] != NULL) {
            Free(p -> next[i]);
        }
    }
    free(p);
}

char s[15];
int main() {
    while(gets(s)) {
        if(strlen(s) == 0) {
            break;
        }
        add(s);
    }
    while(EOF != scanf("%s",s)) {
        printf("%d
",cal(s));
    }
    Node *p = &root;
//    free(p);
}