归途与征程

题目描述

归途与征程_第1张图片

简单的思路

把B串倍长,那么新的串任意一个长度为M的子串都是B的循环同构串。
按照*号把A串分成了若干段,然后每一段都与B串做一次KMP匹配。
具体的,设一个bz[i,j]表示B串第i个位置为开头是否能与第j段匹配。
那么统计答案就是枚举循环同构串的头部,不断往后跑。
当然,如果A串的头部或尾部不是 ,那么要把它弄成 号。
跑的速度很慢达到m^2,所以预处理suc数组表示每个位置后第一个可以匹配的位置然后就是nm的了。

#include
#include
#include
#define fo(i,a,b) for(i=a;i<=b;i++)
#define fd(i,a,b) for(i=a;i>=b;i--)
using namespace std;
const int maxn=100+10,maxm=100000+10;
char a[maxn],s[maxn],b[maxm*2];
int next[maxm*2],suc[maxm*2][maxn],len[maxn];
bool bz[maxm*2][maxn];
int i,j,k,l,t,n,m,ans,top;
bool czy,xdl;
void get(){
    int i,j=0;
    fo(i,2,l){
        while (j&&s[j+1]!=s[i]) j=next[j];
        if (s[j+1]==s[i]) j++;
        next[i]=j;
    }
    j=0;
    fo(i,1,m*2){
        while (j&&s[j+1]!=b[i]) j=next[j];
        if (s[j+1]==b[i]) j++;
        if (j==l){
            bz[i-l+1][top]=1;
            j=next[j];
        }
    }
    suc[m*2+1][top]=m*2+1;
    fd(i,m*2,1)
        if (bz[i][top]) suc[i][top]=i;else suc[i][top]=suc[i+1][top];
}
bool solve(int st){
    int ed=st+m-1,l=1,r=top;
    if (l==r&&!czy&&!xdl){
        if (len[1]!=m) return 0;
        return bz[st][1];
    }
    if (!czy){
        if (!bz[st][1]) return 0;
        l++;
        st+=len[1];
    }
    if (!xdl){
        if (!bz[ed-len[top]+1][top]) return 0;
        r--;
        ed-=len[top];
    }
    int i;
    fo(i,l,r){
        st=suc[st][i];
        if (st+len[i]-1>ed) return 0;
        st+=len[i];
    }
    return 1;
}
bool has_solution(){
    int i,cnt=0;
    fo(i,1,n)
        if (a[i]!='*') cnt++;
    if (cnt>m) return 0;else return 1;
}
int main(){
    //freopen("t3.in","r",stdin);
    scanf("%s",a+1);
    n=strlen(a+1);
    if (a[1]=='*') czy=1;
    if (a[n]=='*') xdl=1;
    scanf("%s",b+1);
    m=strlen(b+1);
    if (!has_solution()){
        printf("0\n");
        return 0;
    }
    fo(i,1,m) b[i+m]=b[i];
    i=1;
    while (i<=n){
        l=0;
        while (i<=n){
            if (a[i]=='*') break;
            s[++l]=a[i];
            i++;
        }
        if (!l){
            i++;
            continue;
        }
        len[++top]=l;
        get();
    }
    ans=0;
    fo(i,1,m)
        ans+=solve(i);
    printf("%d\n",ans);
}

你可能感兴趣的:(模拟,KMP)