【滚动训练】JOYOI 1068 STR （基于Hash+二分的LCP求法）

题意

给你两个串A，B，可以得到从A的任意位开始的子串和B匹配的长度。
给定K个询问，对于每个询问给定一个x，求出匹配长度恰为x的位置有多少个。
N，M，K<=200000

题解

KMP是正解，这里就用Hash+二分水一下，复杂度$nlogn$。
具体做法如下：
首先将A、B两个串全部用同一种Hash方法进行Hash，然后需要统计匹配$i$个字符有多少种情况。 不难想到匹配满足单调性：枚举A串的起点，二分匹配的长度mid，若$A[i,i+mid-1]$ 与 $B[1,mid]$ 匹配，向右二分匹配的位置，否则向左二分匹配的位置。最后统计结果即可。

代码

#include<bits/stdc++.h>
using namespace std;
typedef unsigned long long ull;
const int nmax = 2e5 + 100;
const int INF = 0x3f3f3f3f;
const ull p = 67;
int lenstr,lentar;
int ans[nmax],q,x;
char str[nmax],tar[nmax];
ull hashstr[nmax],hashtar[nmax],pp[nmax] = {0,p};
inline ull hashsub(int l, int r){
    return hashstr[r] - hashstr[l-1] * pp[r-l+1];
}
int main() {
    for(int i = 2;i<nmax;++i) pp[i] = pp[i-1] * p;
    scanf("%d %d %d",&lenstr,&lentar,&q);
    scanf("%s %s",str+1,tar+1); hashstr[1] = str[1], hashtar[1] = tar[1];
    for(int i = 2;i<=lenstr;++i) hashstr[i] = hashstr[i-1] * p + str[i];
    for(int i = 2;i<=lentar;++i) hashtar[i] = hashtar[i-1] * p + tar[i];
    for(int i = 1;i<=lenstr;++i){
        int l = 1, r = min(lentar,lenstr - i + 1), mid,pos = 0;
        while(l <= r){
            mid = (l+r)>>1;
            if(hashsub(i,i+mid-1)== hashtar[mid]) pos = mid, l = mid + 1;
            else r = mid-1;
        }
        ans[pos] ++;
    }
    while(q--){scanf("%d",&x);printf("%d\n",ans[x]);}
    return 0;
}