HDU 2846 Repository(字典树变形题)

题目链接

大意：给定你一些商品的信息save（字符串），然后输入需要查询的字符串str，如果存在一个字符串save使得str是save的子串。

 

思路：字典树的变形，如存在"abcd"，则"bcd", "cd", "d",都需要插入字典树中。如果存在"abab"的话，有可能"ab"会重复计数，所以需要一个特定的序列号来区分是否是同一序列中的子串。

我们可以定义一个value，若value相同，则跳过，若不同，则+1后更新value。

AC代码如下：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<sstream>
#include<fstream>
#include<vector>
#include<map>
#include<stack>
#include<list>
#include<set>
#include<queue>
#define LL long long
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1 | 1
using namespace std;
struct node
{
    int num,id;
    node *next[26];
    node() { num=0;id=0;memset(next,0,sizeof(next));}
};
node *p,*root;
void Insert(string s,int cnt)
{
    p=root;
    for(int i=0;i<s.size();i++)
    {
        int id=s[i]-'a';
        if(p->next[id]==0)
        {
            p->next[id]=new node();
            p=p->next[id];
            p->id=cnt;
            p->num++;
        }
        else
        {
            p=p->next[id];
            if(p->id!=cnt)
            {
                p->id=cnt;
                p->num++;
            }
        }
    }
}
int Query(string s)
{
    p=root;
    for(int i=0;i<s.size();i++)
    {
        int id=s[i]-'a';
        if(p->next[id]==0) return 0;
        p=p->next[id];
    }
    return p->num;
}
void Delete(node *rt)
{
    for(int i=0;i<26;i++)
    {
        if(rt->next[i])
        {
            Delete(rt->next[i]);
            delete(rt->next[i]);
        }
    }
}
int main()
{
    int n;
    string s;
    cin.sync_with_stdio(false);
    while(cin>>n)
    {
        root=new node();
        for(int i=0;i<n;i++)
        {
            cin>>s;
            for(int j=0;j<s.size();j++)
                Insert(s.substr(j),i);
        }
        int m;
        cin>>m;
        for(int i=0;i<m;i++)
        {
            cin>>s;
            cout<<Query(s)<<endl;
        }
        Delete(root);
    }
    return 0;
}