题意:定义两个字符串前后相加能够形成AA形式的字符串即认为两个字符串匹配成功,例如abcdabcd中,A=”abcd“,输入给出n个字符串,输出匹配成功的字符串的对数
思路:可以对每个字符串进行哈希,然后寻找每个字符串是否拥有相同前后缀,若有,则相同前后缀中间部分的子串即为可以与当前字符串匹配成功的字符串,然后再寻找与此子串相同的字符串数量即可,最后再查找一下相同字符串的数量,因为两个相同的字符串也可以匹配成功,此处可以用排列组合公式n(n-1)/2,由于本题会卡字符串冲突,所以需要双哈希处理冲突
代码:
#include<bits/stdc++.h>
#define endl '\n'
#define ll long long
#define Mirai ios::sync_with_stdio(0),cin.tie(0),cout.tie(0);
#define pb push_back
#define INF 0x3f3f3f3f
#define int unsigned long long
//#define int long long
using namespace std;
typedef pair<int,int> PII;
const int mod1=1e9+7,mod2=1e9+9,N=4e5+10,base=131;
pair<int,int> res[N];
map<pair<int,int>,int> mp;
int h1[N],h2[N],p1[N],p2[N];
string s;
void hash1(string s)
{
h1[0]=0;
p1[0]=1;
for(int i=1;i<=s.size();i++)
{
p1[i]=p1[i-1]*base%mod1;
h1[i]=(h1[i-1]*base%mod1+(s[i-1]-'a'+1))%mod1;
}
}
void hash2(string s)
{
h2[0]=0;
p2[0]=1;
for(int i=1;i<=s.size();i++)
{
p2[i]=p2[i-1]*base%mod2;
h2[i]=(h2[i-1]*base%mod2+(s[i-1]-'a'+1))%mod2;
}
}
int get1(int l,int r)
{
return (h1[r]-h1[l-1]*p1[r-l+1]%mod1+mod1)%mod1;
}
int get2(int l,int r)
{
return (h2[r]-h2[l-1]*p2[r-l+1]%mod2+mod2)%mod2;
}
void solve()
{
int n;
cin>>n;
int pos=0;
int ans=0;
for(int i=0;i<n;i++)
{
cin>>s;
hash1(s),hash2(s);
int len=s.size();
mp[{h1[len],h2[len]}]++;//记录当前字符串出现的次数
for(int i=1;i+i<len;i++)
{
if(get1(1,i)==get1(len-i+1,len)&&get2(1,i)==get2(len-i+1,len))//如果找到相同前后缀
{
res[pos++]={get1(i+1,len-i),get2(i+1,len-i)};//将相同前后缀的中间部分的哈希值加入结果中
}
}
}
for(int i=0;i<pos;i++)
{
ans+=mp[res[i]];//统计结果出现的次数
}
for(auto it:mp)
{
ans+=it.second*(it.second-1)/2;//统计相同字符串出现的次数
}
cout<<ans<<endl;
}
signed main()
{
Mirai;
int T=1;
//cin>>T;
while(T--)
{
solve();
}
return 0;
}