题目大意:手机键盘中与数字2相对应的字母有a,b,c;3相对应的字母有d,e,f。给出一些数字串如34,和一些小写字母串。求小写字母对应的数字串出现的次数。字符串abc对应的数字串是111,dh对应的数字串是34。则小写字符串中111出现一次,34出现一次。
题解:可以直接字典树,或者暴力做,毕竟数据比较弱。。。一开始理解错题意,便把数字按照所有转化一遍,再去用字典树,结果超时。。。直接将数字串建树,字母串按对应转化为数字串匹配
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <algorithm>
#include <cmath>
#include <set>
using namespace std;
const int maxn = 200005;
struct tree
{
int id; //最后一次经过此结点的商品ID
int num; //记录包含该结点的单词个数
tree *next[26];
}*root;
char str[11],s[11];
int num[5050];
void creat(char s[],int k)
{
int len = strlen(s);
tree *p = root,*q;
for(int i = 0; i < len; i++)
{
int x = s[i] - '0';
if(p->next[x] == NULL)
{
q = (tree*)malloc(sizeof(tree));
q->num = 0;
q->id = -1;
for(int j = 0; j < 26; j++)
q->next[j] = NULL;
p->next[x] = q;
}
p = p->next[x];
}
p->num = 1;
p->id = k;
}
void Find(char s[])
{
int len = strlen(s);
tree *p = root;
for(int i = 0; i < len; i++)
{
int x = s[i] - '0';
if(p->next[x])
p = p->next[x];
else
return;
}
if(p->id >= 0)
num[p->id]++;
}
void del(tree *root)
{
for(int i = 0; i < 26; i++)
{
if(root->next[i])
del(root->next[i]);
}
free(root);
}
int main()
{
int n,m,t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
root=(tree *)malloc(sizeof(tree));
for(int i = 0; i < 26; i++)
root->next[i] = NULL;
root->num = 0;
root->id = -1;
memset(num,0,sizeof(num));
for(int i=0; i<n; i++)
{
scanf("%s",s);
creat(s,i);
}
for(int i = 0; i < m; i++)
{
scanf("%s",str);
int len = strlen(str);
for(int j = 0; j < len; j++)
{
if(str[j] < 'd')
str[j] = '2';
else if(str[j] < 'g')
str[j] = '3';
else if(str[j] < 'j')
str[j] = '4';
else if(str[j] < 'm')
str[j] = '5';
else if(str[j] < 'p')
str[j] = '6';
else if(str[j] < 't')
str[j] = '7';
else if(str[j] < 'w')
str[j] = '8';
else
str[j] = '9';
}
Find(str);
}
for(int i = 0; i < n; i++)
printf("%d\n",num[i]);
del(root);
}
return 0;
}