洛谷P3763 [TJOI2017]DNA

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string>
using namespace std;
typedef unsigned long long ull; 
const int N=100005,p=131;//p进制 hash 
int T,n,ans,len1,len2;
ull f[N],key1[N],key2[N];
char s1[N],s2[N];
void init(){//字符串hash 
    memset(key1,0,sizeof(key1));//可以通过对二分和判定答案时边界的控制避免每次memset的耗时 
    memset(key2,0,sizeof(key2));
    for(int i=1;i<=len1;++i){
        key1[i]=key1[i-1]*p+(ull)s1[i];
    }
    for(int i=1;i<=len2;++i){
        key2[i]=key2[i-1]*p+(ull)s2[i];
    }
}
ull query(int flag,int pos1,int pos2){//查询两个字符串中pos1到pos2的hash值 
    if(flag==1) return key1[pos2]-key1[pos1-1]*f[pos2-pos1+1];
    else return key2[pos2]-key2[pos1-1]*f[pos2-pos1+1];
}
int lcp(int pos1,int pos2){//二分+hash求s1串中以pos1位置开始和s2串中以pos2位置开始的lcp 
    int l=0,r=min(len1,len2);//r设的较大 貌似可以通过计算传入r 可以避免每次求hash值时memset的时间 
    while(l<r){
        int mid=(l+r+1)>>1;//丑陋的二分写法... 
        if(mid>0&&query(1,pos1,pos1+mid-1)==query(2,pos2,pos2+mid-1)) l=mid;
        else r=mid-1;
    }
    return l;
}
bool check(int pos){
    int cnt=0,pos2=1,t;
    while(cnt<3){//跳三次还不同说明三处以上不同 
        t=lcp(pos,pos2);//lcp 
        pos+=t+1;//这一位不同 两个指针跳到lcp后一位 
        pos2+=t+1;
        if(pos2>len2) return true;
        ++cnt;
    }
    t=lcp(pos,pos2);//最后判断一下 可能跳三次以后到达s2最后一位 恰巧相等 
    pos+=t,pos2+=t;
    return pos2>len2;
}
int main(){
    f[0]=1;
    for(int i=1;i<=N;++i) f[i]=f[i-1]*p;//预处理p的幂 减少计算耗时 
    scanf("%d",&T);
    while(T--){
        scanf("%s",s1+1);
        scanf("%s",s2+1);
        len1=strlen(s1+1),len2=strlen(s2+1);
        if(len1<len2) {//貌似不用解释了？？？ 
            puts("0");
            continue;
        }
        init();//hash 预处理 
        ans=0;
        for(int i=1;i<=len1-len2+1;++i)//枚举每一位作为起点 
            if(check(i)) ++ans;
        printf("%d\n",ans);
    }
    return 0;
}