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

题意:
中问题。
题解:
字典树的模板题,我是看这位大佬学习的字典树:
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));
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值