题意:两个字符串a,b,一个整数n。要求在b中找前缀,其中该前缀在字符串a中至少出现过n次,求最长前缀是多少
题解:对于前缀的长度我们可以二分答案(仔细想想),当前前缀长度mid可以,那么l=mid+1,ans=mid,如果不可以则r=mid-1,从而找到该长度,但是对于每一个mid我都需要找当前mid长度的前缀出现多少次,那么肯定会T,那么这里就需要一个字符串l~r范围内的hash了,对于每个起点往后找mid长度,看当前的hash值是否跟给定b字符串中长度为mid的hash值相同,相同每次num++,看num的个数是否大于等于n就好了
代码:
#include<bits/stdc++.h>
#define ull unsigned long long
#define N 100010
using namespace std;
const ull P=1311;
char a[N],b[N];
int n,lena,lenb,l,r,ans=-1;
ull c[N],hasa[N],hasb[N];
bool check(int mid)
{
int num=0;
for(int i=1;i<=lena-mid+1;i++)
{
ull k=hasa[i+mid-1]-hasa[i-1]*c[mid];
if(k==hasb[mid])
{
num++;
if(num>=n)return 1;
}
}
return 0;
}
int main()
{
fgets(a,N,stdin);
fgets(b,N,stdin);
scanf("%d",&n);
c[0]=1;
for(int i=1;i<=N;i++)
c[i]=c[i-1]*P;
lena=strlen(a);
hasa[0]=0;
for(int i=0;i<lena;i++)
hasa[i+1]=hasa[i]*P+a[i];
lenb=strlen(b);
hasb[0]=0;
for(int i=0;i<lenb;i++)
hasb[i+1]=hasb[i]*P+b[i];
l=1;r=lenb;
while(l<=r)
{
int mid=(l+r)>>1;
if(check(mid))
{
l=mid+1;
ans=mid;
}
else r=mid-1;
}
if(ans==-1)printf("IMPOSSIBLE");
else
{
for(int i=0;i<ans;i++)
printf("%c",b[i]);
}
return 0;
}