hdu 2846 统计给定字符串是多少字符串的子串

不念不忘少年蓝@ 2022-08-05 11:21 130阅读 0赞

**题目链接：[http://acm.hdu.edu.cn/showproblem.php?pid=2846][http_acm.hdu.edu.cn_showproblem.php_pid_2846]**

给定一些字符串，约10000个，给定q的询问，每次询问有一个字符串，判断这个字符串是多少字符串的子串。

字典树处理，把给定字符串的每个后缀都插入字典树，要注意不要把相同的子串重复计数了，因此设置一个id标记，判断之前一个计数是不是当前字符串计数的。

另外字典树的数组要设置的大一些，，，不然会TLE。

#include<bitset>  //G++ 109 ms
    #include<map>
    #include<vector>
    #include<cstdio>
    #include<iostream>
    #include<cstring>
    #include<string>
    #include<algorithm>
    #include<cmath>
    #include<stack>
    #include<queue>
    #include<set>
    #define inf 0x3f3f3f3f
    #define mem(a,x) memset(a,x,sizeof(a))
    
    using namespace std;
    
    typedef long long ll;
    typedef pair<int,int> pii;
    
    inline int in()
    {
        int res=0;char c;
        while((c=getchar())<'0' || c>'9');
        while(c>='0' && c<='9')res=res*10+c-'0',c=getchar();
        return res;
    }
    const int N=1000010;
    
    struct trie
    {
        int next[27];
        bool flag;
        int num,id;
        void init()
        {
            flag=0;
            num=0;
            id=-inf;
            mem(next,-1);
        }
    }trie[N];
    
    char word[22];
    int tot;
    
    void insert(int i,int id,int n)
    {
    
        int now=0,tmp,next;
        for(;i<n;i++)
        {
            tmp=word[i]-'a';
            next=trie[now].next[tmp];
            if(next==-1)
            {
                next = ++tot;
                trie[next].init();
                trie[now].next[tmp]=next;
            }
            if(trie[now].id!=id)
            {
                trie[now].id=id;
                trie[now].num++;
            }
            now=next;
        }
        if(trie[now].id!=id)
        {
            trie[now].id=id;
            trie[now].num++;
        }
        trie[now].flag=1;
    
    }
    
    int query()
    {
        int n=strlen(word);
        int now=0,tmp,next;
        for(int i=0;i<n;i++)
        {
            tmp=word[i]-'a';
            next=trie[now].next[tmp];
            if(next==-1) return 0;
            now=next;
        }
        return trie[now].num;
    }
    
    int main()
    {
        int T=in();
        tot=0;
    
        trie[0].init();
    
        while(T--)
        {
            gets(word);
            int n=strlen(word);
            for(int i=0;i<n;i++)
                insert(i,T,n);
        }
        int q=in();
        while(q--)
        {
            gets(word);
            printf("%d\n",query());
        }
        return 0;
    }

[http_acm.hdu.edu.cn_showproblem.php_pid_2846]: http://acm.hdu.edu.cn/showproblem.php?pid=2846