hihoCoder 1260 String Problem I

最新推荐文章于 2022-04-11 16:59:16 发布

AC_way

最新推荐文章于 2022-04-11 16:59:16 发布

阅读量1.6k

点赞数

分类专栏： ACM-字典树文章标签：算法编程 acm algorithm acmicpc

本文链接：https://blog.csdn.net/caduca/article/details/50414309

版权

ACM-字典树专栏收录该内容

5 篇文章 0 订阅

订阅专栏

时间限制: 10000ms

单点时限: 1000ms

内存限制: 256MB

描述

我们有一个字符串集合S，其中有N个两两不同的字符串。

还有M个询问，每个询问给出一个字符串w，求有多少S中的字符串可以由w添加恰好一个字母得到。

字母可以添加在包括开头结尾在内的任意位置，比如在"abc"中添加"x"，就可能得到"xabc", "axbc", "abxc", "abcx".这4种串。

输入

第一行两个数N和M，表示集合S中字符串的数量和询问的数量。

接下来N行，其中第i行给出S中第i个字符串。

接下来M行，其中第i行给出第i个询问串。

所有字符串只由小写字母构成。

数据范围：

N,M<=10000。

S中字符串长度和<=100000。

所有询问中字符串长度和<=100000。

输出

对每个询问输出一个数表示答案。

样例输入

3 3
tourist
petr
rng
toosimple
rg
ptr

样例输出

解题思路：

用字典树做的，将第1个字符串集建树，对第2个字符串，去树上匹配，要在树上跳过1个节点的才是满足条件的，于是就递归枚举跳过的节点就行了。

代码：

#include <iostream>
#include <cstring>
#include <algorithm>
#include <cstdio>
#include <string>
#include <vector>
#include <map>
#include <set>
using namespace std;
const int maxn=100000+100;
char s[maxn];
const int Max=27;
struct node
{
    int sign;
    int next[Max];
} a[maxn];
set<int> se;
int cur=0;
void insert(char *s)
{
    int len,ans;
    int p=0;
    len=strlen(s);
    for(int i=0; i<len; i++)
    {
        ans=s[i]-'a';
        if(a[p].next[ans]!=0)
        {
            p=a[p].next[ans];
        }
        else
        {
            a[p].next[ans]=++cur;
            a[cur].sign=0;
            p=a[p].next[ans];
        }
    }
    a[p].sign++;
}
//int ans;
int len;
void find(int sign,int p,int x)
{
    if(x==len&&sign&&a[p].sign)
    {
       // ans+=a[p].sign;
        se.insert(p);
        //cout<<ans<<endl;
        return;
    }
    if(sign==0&&x<=len)
    {
        for(int i=0; i<26; i++)
        {
            if(a[p].next[i]>0)
            {
                int tp=a[p].next[i];
                find(1,tp,x);
            }
        }
    }
    if(x<len)
    {
       int ne=s[x]-'a';
       if(a[p].next[ne]>0)
       {
           int tp=a[p].next[ne];
           find(sign,tp,x+1);
       }
    }
}
int main()
{
    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        for(int i=0; i<maxn; i++)
        {
            cur=0;
            a[i].sign=0;
            memset(a[i].next,0,sizeof(a[i].next));
        }
        for(int i=0; i<n; i++)
        {
            scanf("%s",s);
            insert(s);
        }
        for(int i=0; i<m; i++)
        {
            se.clear();
            scanf("%s",s);
            len=strlen(s);
            //ans=0;
            find(0,0,0);
            printf("%d\n",se.size());
        }

    }
    return 0;
}