题目描述
给定 n 个模式串 s1,s2,…,sn 和 q 次询问,每次询问给定一个文本串 ti,请回答 s1∼sn 中有多少个字符串 sj 满足 ti 是 sj 的前缀。
一个字符串 t 是 s 的前缀当且仅当从 s 的末尾删去若干个(可以为 0 个)连续的字符后与 t 相同。
输入的字符串大小敏感。例如,字符串
Fusu和字符串fusu不同。输入格式
本题单测试点内有多组测试数据。
输入的第一行是一个整数,表示数据组数 T。
对于每组数据,格式如下:
第一行是两个整数,分别表示模式串的个数 n 和询问的个数 q。
接下来 n 行,每行一个字符串,表示一个模式串。
接下来 q 行,每行一个字符串,表示一次询问。输出格式
按照输入的顺序依次输出各测试数据的答案。
对于每次询问,输出一行一个整数表示答案。输入输出样例
输入 #1复制
3 3 3 fusufusu fusu anguei fusu anguei kkksc 5 2 fusu Fusu AFakeFusu afakefusu fusuisnotfake Fusu fusu 1 1 998244353 9输出 #1复制
2 1 0 1 2 1说明/提示
数据规模与约定
对于全部的测试点,保证 1≤T,n,q≤105,且输入字符串的总长度不超过 3×106。输入的字符串只含大小写字母和数字,且不含空串。
说明
std 的 IO 使用的是关闭同步后的 cin/cout,本题不卡常。
#include<bits/stdc++.h> using namespace std; const int N=3e6+10; int tr[N][62]; int p[N]; int idx; int T,n,q; string s; int get_num(char x) { if(x>='a'&&x<='z') return x-'a'; else if(x>='A'&&x<='Z') return x-'A'+26; else return x-'0'+52; } void insert(string& s) { int cur=0; p[cur]++; for(auto x:s) { int path=get_num(x); if(tr[cur][path]==0) tr[cur][path]=++idx; cur = tr[cur][path]; p[cur]++; } } int find_pre(string& s) { int cur =0; for(auto x: s) { int path=get_num(x); if(tr[cur][path]==0) return 0; cur=tr[cur][path]; } return p[cur]; } int main() { cin>>T; while(T--) { //清空 for(int i=0;i<=idx;i++) { for(int j=0;j<62;j++) { tr[i][j]=0; } } for(int i=0;i<=idx;i++) p[i]=0; idx=0; cin>>n>>q; while(n--) { cin>>s; insert(s); } while(q--) { cin>>s; cout<<find_pre(s)<<endl; } } }