Birthday Cake

最新推荐文章于 2024-08-13 14:56:59 发布

_Mirai

最新推荐文章于 2024-08-13 14:56:59 发布

阅读量42

点赞数 1

分类专栏：字符串哈希文章标签：算法

本文链接：https://blog.csdn.net/qq_73051419/article/details/130855033

版权

字符串哈希专栏收录该内容

2 篇文章 0 订阅

订阅专栏

该文描述了一个编程问题，涉及字符串处理。给定n个字符串，目标是找出所有能组成AA形式的字符串配对。解决方案是使用哈希函数检查每个字符串的前后缀是否相同，并存储中间子串的哈希值，通过计算哈希冲突来确定匹配计数。

摘要由CSDN通过智能技术生成

题目链接

题意：定义两个字符串前后相加能够形成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;
}