HDU1251 字典树模板题，数组版和指针版

最新推荐文章于 2023-01-19 18:46:53 发布

原创最新推荐文章于 2023-01-19 18:46:53 发布 · 395 阅读

0 ·

CC 4.0 BY-SA版权

字典树专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种使用字典树解决字符串查找问题的方法，并提供了两种实现方式：数组版和指针版。通过实例演示了如何构建字典树并进行字符串匹配。

题意：
中问题。
题解：
字典树的模板题，我是看这位大佬学习的字典树：
https://www.cnblogs.com/TheRoadToTheGold/p/6290732.html
字典树的最大时间复杂度为O(n),n代表字符串的长度。
这道题主要用来练练手，看看模板是否有问题，这道题我分别写了数组版的和指针版的，神奇的是指针版的在G++会超内存，而C++过了。。。不知道哪里有问题，求大佬讲解一下，不甚感激。

数组版

//数组版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=500000+7;
int trie[MAXN][26];
int vis[MAXN];
int tot=1;
void insert(char *s)
{
    int rt=0;
    for(int i=0;s[i];i++)
    {
        int x=s[i]-'a';
        if(trie[rt][x]==0)
        {
            trie[rt][x]=tot++;
        }
        rt=trie[rt][x];
        vis[rt]++;
    }
}
int find(char *s)
{
    int rt=0;
    for(int i=0;s[i];i++)
    {
        int x=s[i]-'a';
        if(trie[rt][x]==0)
        return 0;
        rt=trie[rt][x];
    }
    return vis[rt];
}
int main()
{
    memset(trie,0,sizeof(trie));
    memset(vis,0,sizeof(vis));
    char s[15];
    while(gets(s)&&s[0]!='\0')
    insert(s);
    while(~scanf("%s",s))
    {
        printf("%d\n",find(s));
    }
}

指针版

//指针版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
    int cnt;
    node *next[26];
}*root;
node *build()
{
    node *k=new(node);
    k->cnt=0;
    memset(k->next,0,sizeof(k->next));
    return k;
}
void insert(char *s)
{
    node *r=root;
    for(int i=0;s[i];i++)//这样的写法也是可以的，因为s[i]=='\0'的时候会跳出。
    {
        int x=s[i]-'a';
        if(r->next[x]==0) r->next[x]=build();
        r=r->next[x];
        r->cnt++; 
    } 
}
int find(char *s)
{
    node *r=root;
    for(int i=0;s[i];i++)
    {
        int x=s[i]-'a';
        if(r->next[x]==0) return 0;
        r=r->next[x];
    }
    return r->cnt;
}
int main()
{
    root=build();
    char s[15];
    while(gets(s)&&s[0]!='\0')
    {
        insert(s); 
    }
    while(~scanf("%s",s))
    {
        printf("%d\n",find(s));
    }
}
//这道题的指针版要用C++才能过。。。。不知道为什么。。

还有一个指针版

//另外一个指针版
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
struct node
{
    int cnt;
    node *next[26];
    node(){
        cnt=0;
        for(int i=0;i<26;i++)
        next[i]=0;
    }
}*root;
void insert(char *s)
{
    node *r=root;
    for(int i=0;s[i];i++)//这样的写法也是可以的，因为s[i]=='\0'的时候会跳出。
    {
        int x=s[i]-'a';
        if(r->next[x]==0) r->next[x]=new(node);
        r=r->next[x];
        r->cnt++; 
    } 
}
int find(char *s)
{
    node *r=root;
    for(int i=0;s[i];i++)
    {
        int x=s[i]-'a';
        if(r->next[x]==0) return 0;
        r=r->next[x];
    }
    return r->cnt;
}
int main()
{
    root=new(node);
    char s[15];
    while(gets(s)&&s[0]!='\0')
    {
        insert(s); 
    }
    while(~scanf("%s",s))
    {
        printf("%d\n",find(s));
    }
}