hihoCoder 1260 String Problem I

来源：互联网发布：临沂知豆租赁电话 400 编辑：程序博客网时间：2024/05/01 02:42

#1260 : String Problem I

时间限制: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 3touristpetrrngtoosimplergptr

样例输出

解题思路：Trie+DFS，提供几组数据

3 3

aba bab aab

ab ba aa

3 2 1

3 1

aab aba abb

ab

#include <iostream>#include <cstdio>#include <cstring>#include <string>#include <algorithm>#include <set>#include <map>#include <list>#include <queue>#include <stack>#include <deque>#include <vector>#include <bitset>#include <cmath>#include <utility>#define Maxn 100005#define Maxm 1000005#define lowbit(x) x&(-x)#define lson l,m,rt<<1#define rson m+1,r,rt<<1|1#define PI acos(-1.0)#define make_pair MP#define LL long long #define Inf (1LL<<62)#define inf 0x3f3f3f3f#define re freopen("in.txt","r",stdin)#define wr freopen("out.txt","w",stdout)using namespace std;struct TrieNode{    int cnt;    TrieNode *next[26];    TrieNode()    {cnt=0;        memset(next,0,sizeof(next));    }};TrieNode *root=NULL;void CreatTree(char *s,int len){    TrieNode *p=root;    TrieNode *temp;    for(int i=0;i<len;i++)    {        if(p->next[s[i]-'a']==NULL)        {            temp=new TrieNode;            p->next[s[i]-'a']=temp;        }        p=p->next[s[i]-'a'];p->cnt++;    }}int ans;void Search(TrieNode *node,char *s,int pos,int len,int flag){if(pos==len){ans+=node->cnt;return ;}    for(int i=0;i<26;i++){if(node->next[i]!=NULL){if(s[pos]=='a'+i)Search(node->next[i],s,pos+1,len,flag);else if(flag==1)Search(node->next[i],s,pos,len,flag-1);}}return ;}void Delete(TrieNode *node){    for(int i=0;i<26;i++)    {        if(node->next[i])            Delete(node->next[i]);        delete node->next[i];        node->next[i]=0;    }}int main(){int m,n,index;bool visit[100005];map<string,bool> vis;root=new TrieNode;char str1[10005],str2[10005];while(~scanf("%d %d",&m,&n)){getchar();memset(visit,false,sizeof(visit));for(int i=0;i<m;i++){scanf("%s",str1);visit[strlen(str1)]=true;CreatTree(str1,strlen(str1));}for(int i=0;i<n;i++){ans=0;scanf("%s",str2);vis.clear();if(!visit[strlen(str2)+1]){puts("0");continue;}int len=strlen(str2);Search(root,str2,0,len,1);printf("%d\n",ans);}Delete(root);}return 0;}

0 0