寻找子串位置<codevs>

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<cstring>
using namespace std;
string s1,s2;
int len1,len2;
int f[200000];
int main(){
    cin>>s1>>s2;
    len1=s1.size(),len2=s2.size();
    f[1]=f[0]=0;
    for(int i=1;i<len2;i++){
        int j=f[i];
        while(j&&s2[i]!=s2[j]) j=f[j];
        if(s2[i]==s2[j]) f[i+1]=j+1;
        else f[i+1]=0;
    }
    int j=0;
    for(int i=0;i<=len1;i++){
        while(j&&s1[i]!=s2[j]) j=f[j];
        if(s1[i]==s2[j]) j++;
        if(j==len2) {printf("%d",i-j+2);break;}
    }
    return 0;
}