题目链接:传送门
描述
给定 $N$ 个字符串 $S_1,S_2,\cdots,S_N$,接下来进行 $M$ 次询问,每次询问给定一个字符串 $T$,求 $S_1 \sim S_N$ 中有多少个字符串是 $T$ 的前缀。输入字符串的总长度不超过 $10^6$,仅包含小写字母。
输入格式
第一行两个整数 $N,M$。接下来 $N$ 行每行一个字符串 $S_i$。接下来 $M$ 行每行一个字符串表示询问。
输出格式
对于每个询问,输出一个整数表示答案
样例输入
3 2
ab
bc
abc
abc
efg
样例输出
2
0
模板(一个最最普通的字典树模板):
namespace Trie { const int SIZE=1e6+10; int sz; struct TrieNode { int ed; int nxt[26]; }trie[SIZE]; void init() { sz=1; for(int i=0;i<SIZE;i++) { trie[i].ed=0; memset(trie[i].nxt,0,sizeof(trie[i].nxt)); } } void insert(string s) { int p=1; for(int i=0;i<s.size();i++) { int ch=s[i]-'a'; if(trie[p].nxt[ch]==0) trie[p].nxt[ch]=++sz; p=trie[p].nxt[ch]; } trie[p].ed++; } bool search(string s) { int p=1; for(int i=0;i<s.size();i++) { p=trie[p].nxt[s[i]-'a']; if(!p) return 0; } return trie[p].ed; } };
题解:
字典树板子题,对上面的板子稍作修改即可。
AC代码:
#include<bits/stdc++.h> using namespace std; namespace Trie { const int SIZE=1e6+10; int sz; struct TrieNode { int ed; int nxt[26]; }trie[SIZE]; void init() { sz=1; for(int i=0;i<SIZE;i++) { trie[i].ed=0; memset(trie[i].nxt,0,sizeof(trie[i].nxt)); } } void insert(string s) { int p=1; for(int i=0;i<s.size();i++) { int ch=s[i]-'a'; if(trie[p].nxt[ch]==0) trie[p].nxt[ch]=++sz; p=trie[p].nxt[ch]; } trie[p].ed++; } int search(string s) { int res=0; int p=1; for(int i=0;i<s.size();i++) { p=trie[p].nxt[s[i]-'a']; res+=trie[p].ed; if(!p) return res; } return res; } }; int n,m; string s; int main() { cin>>n>>m; Trie::init(); for(int i=1;i<=n;i++) { cin>>s; Trie::insert(s); } for(int i=1;i<=m;i++) { cin>>s; cout<<Trie::search(s)<<endl; } }