codevs1204 寻找子串位置

题目描述 Description

给出字符串a和字符串b，保证b是a的一个子串，请你输出b在a中第一次出现的位置。

输入描述 Input Description

仅一行包含两个字符串a和b

输出描述 Output Description

仅一行一个整数

样例输入 Sample Input

abcd bc

样例输出 Sample Output

2

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
const int maxn = ;
int n,m,next[maxn];
char t[maxn],p[maxn];
void input(){
    cin>>t>>p;
    n = strlen(t);
    m = strlen(p);
}
void make(){
    next[] = ;
    for(int i = ,k = ;i < m;i++){
        while(k >  && p[i] != p[k]) k = next[k-];
        if(p[i] == p[k]) k++;
        next[i] = k;
    }
}
void kmp(){
    for(int i = ,k = ;i < n;i++){
        while(k >  && t[i] != p[k]){
        k = next[k-];
        }

        if(t[i] == p[k]) k++;
        if(k == m){
            cout<<i - m + ;
            return;
        }
    }
}
int main(){
    input();
    make();
    kmp();
    return ;
}