转自:cxlove
题意:有两个字符串集合,从第一个集合中取某个串的非空前缀,从第二个集合中取某个串的非空后缀,拼接成一个串,问有多少个不同的新串。
思路:首先把第一个集合中的串加入到Trie中,然后就可以枚举前缀了,然后就是统计后缀,麻烦的是可能出现重复的。
当我们按长度枚举前缀的时候,可以考虑如果后缀的前缀出现在前缀的后继中,那便不用考虑,后面总会统计到。
这样的话,其实只要比较第一个字母。
如果当前结点P表示一个前缀串,那么如果p -> next[i]为空,说明前缀中不含字母i为后继,那么我们就可以统计以i开头的后缀了。否则总能在p -> next[i]为前缀中统计到。
有一点需要注意的是,如果 p -> next[i]便是我们最终的串,即以p 为前缀时,后缀只有一个字母的话,那就不能把这一个字母拉入到前缀中,因为后缀是非空的。所以再处理下end[i],表示是否存在以单个字母i为后缀的。
至于统计以i开头的不重复后缀的话,可以反向加入到Trie中统计。
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<vector>
#include<cmath>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=100010;
const int maxm=1010;
const int SIGMA_SIZE=26;
int P,S;
char a[maxm][maxm],b[maxm][maxm];
int tail[maxn];
int cnt[30];
LL ans;
struct Trie
{
int ch[maxn][SIGMA_SIZE];
int val[maxn];
int sz;
void clear(){memset(ch[0],0,sizeof(ch[0]));sz=1;}
int idx(char x){return x-'a';}
void insert(char *s)
{
int u=0;
for(int i=0;s[i];i++)
{
int c=idx(s[i]);
if(!ch[u][c])
{
memset(ch[sz],0,sizeof(ch[sz]));
val[sz]=0;
ch[u][c]=sz++;
}
u=ch[u][c];
}
val[u]=1;
}
void cal(int u)
{
for(int i=0;i<SIGMA_SIZE;i++)
{
if(ch[u][i])
{
cnt[i]++;
cal(ch[u][i]);
}
}
}
void solve(int u)
{
for(int i=0;i<SIGMA_SIZE;i++)
{
if(!ch[u][i])
{
if(u)
ans+=cnt[i];
}
else
{
if(u&&tail[i])ans++;
solve(ch[u][i]);
}
}
}
}prefix,suffix;
int main()
{
while(scanf("%d%d",&P,&S)!=EOF,P||S)
{
prefix.clear();
suffix.clear();
memset(cnt,0,sizeof(cnt));
memset(tail,0,sizeof(tail));
for(int i=0;i<P;i++)
{
scanf("%s",a[i]);
prefix.insert(a[i]);
}
for(int i=0;i<S;i++)
{
scanf("%s",b[i]);
reverse(b[i],b[i]+strlen(b[i]));
tail[b[i][0]-'a']=1;
suffix.insert(b[i]);
}
ans=0;
suffix.cal(0);
prefix.solve(0);
cout<<ans<<endl;
}
return 0;
}