HDU 2846 字典树 非前缀子串的计数

这道题目与一般的字典树不同的是要查询的字符串不是已经插入字符串的前缀，同时也不是后缀，只需满足该字符串是其中一个商品的子串即可，考虑到既然要用i到字符串，所以对于每一个商品的字符串插入以每一个字符开头后缀，但是后来出现了一个问题，就是对于以i下标开头的子串同时是以i结尾的前面某一字符串的子串时，会出现重复计数的问题，想过在每一个节点用一个set作为经过该节点的字符串的Id号，很显然，内存超了，后来看网上说既然是要避免同一个商品的字符串出现重复问题，只需在节点标记一下经过此节点的最后的那个商品的标号，标号相同时候，count不变，不同的时候count++，这样便解决了问题。。。

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<string>
using namespace std;
struct Trie_node
{
    Trie_node *next[26];
    int count;
    int laststr;
    Trie_node()
    {
        count=0;
        for(int i=0;i<26;i++)
        {
            next[i]=NULL;
            count=0;
            laststr=-1;
        }
    }
};
void insert(char *s,Trie_node *root,int id)
{
    Trie_node *p=root;
    int index;
    while(*s)
    {
        index=*s-'a';
        if(p->next[index]==NULL)
        {
            p->next[index]=new Trie_node();
        }
        p=p->next[index];
        if((p->laststr)!=id)
        {
            p->count++;
            p->laststr=id;
        }
        s++;
    }
}
int find(char *s,Trie_node *root)
{
    Trie_node *p=root;
    int index;
    while(*s)
    {
        index=*s-'a';
        if(p->next[index]==NULL)return 0;
        p=p->next[index];
        s++;
    }
    return p->count;
}
int main()
{
    char s[30];
    int p,Q,len,i;
    Trie_node *root=new Trie_node();
    scanf("%d",&p);
    while(p--)
    {
        scanf("%s",s);
        len=strlen(s);
        for(i=0;i<len;i++)
        {
            insert(s+i,root,p+1);
        }
    }
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%s",s);
        printf("%d\n",find(s,root));
    }
    return 0;
}