[题目链接] (https://ac.nowcoder.com/acm/contest/1076/G)
分享一道hash题,这题有可以学习的地方。
题目很简单,就是给两个串S, T ;要求删除T串中任意的一个字符,使得T串为S串的子串。如果无解输出-1.有解输出最小的下标。
思路就是预处理S串中T串长度减一的所有子串的hash值存在map里,然后对子串的每个位置进行枚举,找到map中对应的hash值。
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 3e5+10;
char s1[maxn],s2[maxn];
map<pair<ll,ll>,int> mp;
ll base[2] = {37,41} ,mod[2]={1000000007,998244353};
ll p[2][maxn] ,h[2][maxn];
void init(){
p[0][0] = p[1][0] = 1;
for(int i=1;i<maxn;i++)
for(int j=0;j<2;j++)
p[j][i] = p[j][i-1] * base[j] % mod[j] ;
}
ll gets(int l ,int r, int i)
{
if(l>r) return 0;
ll tmp = (h[i][r]-h[i][l-1]*p[i][r-l+1]%mod[i]+mod[i])%mod[i];
return tmp ;
}
int get(char a)
{
if(a>='a'&&a<='z')
return a - 'a' ;
}
int main()
{
init();
scanf("%s%s",s1+1,s2+1);
int n1 = strlen(s1+1) , n2 = strlen(s2+1);
for(int i=1;i<=n1;i++) // 预处理所有S串中的子串
{
for(int j=0;j<2;j++)
h[j][i] = (h[j][i-1] * base[j] % mod[j] + get(s1[i])) % mod[j] ;
if(i>=n2-1)
mp[make_pair(gets(i-n2+2,i,0),gets(i-n2+2,i,1))] = 1;
}
int ans = -1;
for(int i=1;i<=n2;i++)
for(int j=0;j<2;j++)
h[j][i] = (h[j][i-1] * base[j] % mod[j] + get(s2[i])) % mod[j];
for(int i=1;i<=n2;i++) //枚举T串中每个位置
{
int h1[2],h2[2],res[2];
for(int j=0;j<2;j++)
h1[j] = gets(1,i-1,j) , h2[j] = gets(i+1,n2,j);
for(int j=0;j<2;j++) // 结果合并起来 有趣的地方
res[j] = ( h1[j] * p[j][n2-i] % mod[j] + h2[j] ) % mod[j];
if(mp[make_pair(res[0],res[1])]==1){
ans = i ; break;
}
}
printf("%d\n",ans);
}
另一种合并方法:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn = 3e5+10;
char s1[maxn],s2[maxn];
map<pair<ll,ll>,int> mp;
ll base[2] = {37,41} ,mod[2]={1000000007,998244353};
ll p[2][maxn] ,h[2][maxn];
void init(){
p[0][0] = p[1][0] = 1;
for(int i=1;i<maxn;i++)
for(int j=0;j<2;j++)
p[j][i] = p[j][i-1] * base[j] % mod[j] ;
}
ll gets(int l ,int r, int i)
{
if(l>r) return 0;
ll tmp = (h[i][r]-h[i][l-1]*p[i][r-l+1]%mod[i]+mod[i])%mod[i];
return tmp ;
}
int get(char a)
{
if(a>='a'&&a<='z')
return a - 'a' ;
}
int main()
{
init();
scanf("%s%s",s1+1,s2+1);
int n1 = strlen(s1+1) , n2 = strlen(s2+1);
for(int i=1;i<=n1;i++)
{
for(int j=0;j<2;j++)
h[j][i] = (h[j][i-1] * base[j] % mod[j] + get(s1[i])) % mod[j] ;
if(i>=n2-1)
mp[make_pair(gets(i-n2+2,i,0),gets(i-n2+2,i,1))] = 1;
}
int ans = -1;
for(int i=1;i<=n2;i++)
for(int j=0;j<2;j++)
h[j][i] = (h[j][i-1] * base[j] % mod[j] + get(s2[i])) % mod[j];
for(int i=1;i<=n2;i++)
{
int res1 ,res2 ;
int h1,h2,sum1;
sum1 = gets(1,n2,0), h1 = gets(1,i-1,0) , h2 =gets(1,i,0);
res1 = (sum1 - h2 * p[0][n2-i] % mod[0] + h1 * p[0][n2-i] % mod[0] + mod[0]) % mod[0];
sum1 = gets(1,n2,1) , h1 = gets(1,i-1,1) , h2 =gets(1,i,1);
res2 = (sum1 - h2 * p[1][n2-i] % mod[1] + h1 * p[1][n2-i] % mod[1] + mod[1]) % mod[1];
if(mp[make_pair(res1,res2)]==1){
ans = i ; break ;
}
}
printf("%d\n",ans);
}