找一个字符串有多少不同子串(trie树)

感谢铭哥提供的思路:https://blog.csdn.net/DT2131/article/details/54936247

找一个字符串有多少不同子串,利用trie树性质,即为trie树上有多少个不同的节点。

代码:

#include <iostream>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
using namespace std;
char s[100000];
int ans;
//找一个串不同的子串可用tire树经过每次插入从i到len长度的子串,看tire树中新开的节点个数即不同子串个数

struct Tire{
    int next[500000][128];
    int rt,cnt;
    int newnode()
    {
        for(int i=0;i<128;i++)
        {
            next[cnt][i]=-1;
        }
        ans++;
        return cnt++;
    }
    void init()
    {
        cnt=0;
        rt=newnode();
    }
    void insert(int k,int len)
    {
        int now=rt;
        for(int i=k;i<len;i++)
        {
            if(next[now][s[i]]==-1)
            {
                next[now][s[i]]=newnode();
            }
            now=next[now][s[i]];
        }
    }
};
Tire ac;

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",s);
        ac.init();
          ans=0;
        int len=strlen(s);
        for(int i=0;i<len;i++)
        {
            ac.insert(i,len);
        }
        printf("%d\n",ans);
    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值