Gym - 101466E -Text Editor（hash+二分）

题目：Gym - 101466E

题意：两个字符串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;
}