Trie-优快云博客

本文介绍了一种使用Trie树的数据结构来高效地插入和搜索字符串的方法。具体实现了插入功能Insert、精确搜索功能searchTrie及部分匹配搜索功能Behind，并通过示例代码展示了如何利用这些函数处理大量字符串数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<iostream>
#include<cstdlib>
#include<cstring>
#include<string>
#include<cstdio>
#define Max 26
using namespace std;

struct TrieNode{
    int ti;
    TrieNode *next[Max];
    TrieNode(){
        ti=0;
        memset(next,NULL,sizeof(next));
    }
};

void Insert(TrieNode *root,char * str){
    int len=strlen(str);
    TrieNode *p=root;
    for(int i=0;i<len;++i){
        int id=str[i]-'a';
        if(p->next[id]==NULL)
            p->next[id]=new TrieNode();
        p=p->next[id];
        p->ti++;
    }
}

int searchTrie(TrieNode *root,char *str){
    TrieNode *p=root;
    int len=strlen(str);
    for(int i=0;i<len;++i){
        int id=str[i]-'a';
        if(p->next[id]==NULL)
            return 0;
        p=p->next[id];
    }
    return p->ti;
}

int Behind(TrieNode *root,char *str){
    TrieNode *p=root;
    int len=strlen(str);
    int ans=0;
    for(int i=0;i<len;++i){
        int id=str[i]-'a';
        for(int j=id+1;j<Max;++j)
            if(p->next[j]!=NULL)
                ans+=p->next[j]->ti;
        p=p->next[id];
    }
    for(int i=0;i<Max;++i)
        if(p->next[i]!=NULL)
            ans+=p->next[i]->ti;
    return ans;
}

int n,m,ans;
char str[50005];
char now[50006];

int main(){
    TrieNode *root=new TrieNode();
    scanf("%d",&n);
    for(int i=0;i<n;++i){
        scanf("%s",str);
        Insert(root,str);
    }
    scanf("%d",&m);
    for(int i=1;i<=m;++i){
        scanf("%s",str);
        printf("%d\n",searchTrie(root,str));
    }
    delete root;
    return 0;
}

转载于:https://www.cnblogs.com/qdscwyy/p/7792206.html