【Trie】Trie字典树模板 静态指针池、数组写法

7 篇文章 0 订阅

下面是数组写法:

#include <cstdio>
#include <string>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

#define idx(x) x - 'a';
const int MAXN = 1e6;
struct Trie {
    int next[26];
    int val;
}tree[MAXN];
int nxt, T;
char str[MAXN];

int add()
{
    memset(&tree[nxt], 0, sizeof(Trie));
    return nxt++;
}

void Insert(char *s)
{
    int rt = 0, len = strlen(s);
    for(int i = 0; i < len; i++) {
        int c = idx(s[i]);
        if(!tree[rt].next[c]) {
            tree[rt].next[c] = add();
        }
        rt = tree[rt].next[c];
    }
    tree[rt].val++;
}

bool Find(char *s)
{
    int rt = 0, len = strlen(s);
    for(int i = 0; i < len; i++) {
        int c = idx(s[i]);
        if(!tree[rt].next[c])
            return false;
        rt = tree[rt].next[c];
    }
    if(tree[rt].val) return true;
    return false;
}

int main()
{
    memset(&tree[0], 0, sizeof(Trie));
    nxt = 1;
    scanf("%d", &T);
    while(T--) {
        scanf("%s", str);
        Insert(str);
    }
    while(~scanf("%s", str)) {
        if(Find(str)) puts("exist");
        else puts("none");
    }
    return 0;
}

下面是静态指针池写法:

#include <cstdio>
#include <string>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

#define idx(x) x - 'a';
const int MAXN = 1e6;
struct Trie {
	Trie *next[26];
	int val;
}tree[MAXN];
int nxt, T;
char str[MAXN];

Trie *add()
{
    memset(&tree[nxt], 0, sizeof(Trie));
    return &tree[nxt++];
}

void Insert(Trie *rt, char *s)
{
    int len = strlen(s);
    for(int i = 0; i < len; i++) {
        int c = idx(s[i]);
        if(!rt->next[c])
            rt->next[c] = add();
        rt = rt->next[c];
    }
    rt->val++;
}

bool Find(Trie *rt, char *s)
{
    int len = strlen(s);
    for(int i = 0; i < len; i++) {
        int c = idx(s[i]);
        if(!rt->next[c])
            return false;
        rt = rt->next[c];
    }
    if(rt->val)	return true;
    return false;
}

int main()
{
    nxt = 0;
    Trie *root = add();
    scanf("%d", &T);
    while(T--) {
        scanf("%s", str);
        Insert(root, str);
    }
    while(~scanf("%s", str)) {
        if(Find(root, str)) puts("exist");
        else puts("none");
    }
    return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值