hdu 2846 Repository

题目属于判断字符串中是否包含子串的问题.

思路：对于一般的字典树，用来判断前缀，而这里不能直接这么去建树。在建树的时候将字符串X=X1X2....Xn的分别以X1,X2....Xn开头的后缀子串插入到Trie树中，如此一来就可以判断某个字符串是否被包含在另一个字符串当中。当然，这里还有个问题，比如插入了字符串abab，那么当查找字符串ab时就会重复计数，因此需要多设计一个标识以表示在插入"abab"和"ab"时时同一个字符串即可(是同一个字符串就不需要使计数器加1)，因此在Trie树结点中多设计一个商品id来标记。id用来记录最后一个经过此路径上的商品编号，如果要插入的字符串编号同当前节点的编号不同，则计数器加1，并且将当前结点的编号置为要插入的字符串的编号。

#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <set>
using namespace std;
const int maxn = 200005;
struct tree
{
    int id;         //最后一次经过此结点的商品ID
    int num;            //记录包含该结点的单词个数
    tree *next[26];
};
char s[25];
tree *root=(tree *)malloc(sizeof(tree));
void creat(char s[],int k)
{
    int len = strlen(s);
    tree *p = root,*q;
    for(int i = 0; i < len; i++)
    {
        int x = s[i] - 'a';
        if(p->next[x] == NULL)
        {
            q = (tree*)malloc(sizeof(tree));
            q->id = k;
            q->num = 1;
            for(int j = 0; j < 26; j++)
                q->next[j] = NULL;
            p->next[x] = q;
        }
        p = p->next[x];
        if(p->id != k)
        {
            p->id = k;
            p->num++;
        }
    }
}
int Find(char s[])
{
    int len = strlen(s);
    tree *p = root;
    for(int i = 0; i < len; i++)
    {
        int x = s[i] - 'a';
        if(p->next[x])
            p = p->next[x];
        else
            return 0;
    }
    return p->num;
}
void del(tree *root)
{
    for(int i = 0; i < 26; i++)
    {
        if(root->next[i])
            del(root->next[i]);
    }
    free(root);
}
int main()
{
    int n,m;
    for(int i = 0; i < 26; i++)
        root->next[i] = NULL;
    root->num=0;
    root->id=-1;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
        scanf("%s",s);
        int len=strlen(s);
        for(int j=0;j<len;j++)              //将字符串X=X1X2...Xn的分别以X1,X2...Xn开头的后缀字符串插入到Tree树中
            creat(s+j,i);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s",s);
        printf("%d\n",Find(s));
    }
    del(root);
    return 0;
}