题意:给你几串数字,每一串数字都是拼音九键的按键顺序。再给你几个单词,问你从刚才的按键顺序中分别能按出多少个给定的单词。
思路:用map直接将每个字母映射为拼音九键对应的数字,再插入到字典树,插入完后对应位置u的v[u]+1,v[u]表示打到u这个位置能打出的给定的单词的单词数。总体来说还是比较简单的。
AC代码:
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<string>
#include<queue>
#include<vector>
#include<map>
using namespace std;
const int mx=1000005;
const int inf=0x3f3f3f3f;
int num,cas,n,m,T;
map<char,int> mp;
struct node{
int ch[mx][10];
int v[mx];
int sz;
void add(char *s)
{
int u=0,n=strlen(s);
for(int i=0;i<n;i++)
{
int id=mp[s[i]];
if(ch[u][id]==0)
{
ch[u][id]=sz;
v[sz]=0;
memset(ch[sz],0,sizeof(ch[sz]));
sz++;
}
u=ch[u][id];
}
v[u]++;
}
void find(char *s)
{
int u=0,n=strlen(s);
for(int i=0;i<n;i++)
{
int id=s[i]-'0';
if(ch[u][id]==0) {printf("0\n");return;}
u=ch[u][id];
}
printf("%d\n",v[u]);
}
void init(){
memset(ch[0],0,sizeof(ch[0]));
v[0]=0;
sz=1;
mp['a']=2;mp['b']=2;mp['c']=2;mp['d']=3;mp['e']=3;mp['f']=3;
mp['g']=4;mp['h']=4;mp['i']=4;mp['j']=5;mp['k']=5;mp['l']=5;
mp['m']=6;mp['n']=6;mp['o']=6;mp['p']=7;mp['q']=7;mp['r']=7;mp['s']=7;
mp['t']=8;mp['u']=8;mp['v']=8;mp['w']=9;mp['x']=9;mp['y']=9;mp['z']=9;
}
}a;
int main()
{
//ios::sync_with_stdio(false);
cas=0;
char c[5005][10];
char s[65];
while(scanf("%d",&T)!=EOF)
{
while(T--)
{
a.init();
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%s",c[i]);
for(int i=1;i<=m;i++) {
scanf("%s",s);
a.add(s);
}
for(int i=1;i<=n;i++)
a.find(c[i]);
}
}
return 0;
}