题意:给出n个单词,其中首尾字母相同,中间字母任意排列的单词是同一个码,然后给出m个句子,每个句子含有若干个单词,问多少个不同的句子和这个句子有相同的编码。
题解:把每个单词除去首尾的部分按字典序排序,然后插入字典树,val[u]存节点出现次数,在每个单词的最后一个节点数量累加,然后把输入的句子中每个统计到的单词数量乘起来。
用自己写的句子按空格拆分单词的代码一直wa,看了题解用的strtok(str, ” “)函数,可以把一个字符串按分隔符‘ ’拆分成为一组字符串,用这个函数就可以a,我也是醉了。。。
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 130005;
int Next[N][60], val[N], sz, n, m;
char str[10005];
void init() {
memset(Next[0], 0, sizeof(Next[0]));
val[0] = 0;
sz = 1;
}
void insert(char *s, int len) {
int u = 0;
for (int i = 0; i < len; i++) {
int k;
if (s[i] >= 'a' && s[i] <= 'z')
k = s[i] - 'a' + 1;
else
k = s[i] - 'A' + 27;
if (!Next[u][k]) {
memset(Next[sz], 0, sizeof(Next[sz]));
val[sz] = 0;
Next[u][k] = sz++;
}
u = Next[u][k];
}
val[u]++;
}
int query(char *s, int len) {
int u = 0;
for (int i = 0; i < len; i++) {
int k;
if (s[i] >= 'a' && s[i] <= 'z')
k = s[i] - 'a' + 1;
else
k = s[i] - 'A' + 27;
if (!Next[u][k])
return 0;
u = Next[u][k];
}
return val[u];
}
int main() {
int t, cas = 1;
scanf("%d", &t);
while (t--) {
init();
scanf("%d", &n);
for (int i = 0; i < n; i++) {
scanf("%s", str);
int len = strlen(str);
if (len > 2)
sort(str + 1, str + len - 1);
insert(str, len);
}
scanf("%d", &m);
getchar();
printf("Case %d:\n", cas++);
for (int i = 0; i < m; i++) {
long long res = 1;
gets(str);
char *p = strtok(str, " ");
while (p) {
char s[10005];
strcpy(s, p);
int len = strlen(s);
if (len > 2)
sort(s + 1, s + len - 1);
res *= query(s, len);
p = strtok(NULL, " ");
}
printf("%lld\n", res);
}
}
return 0;
}