Поиск подстроки - C++ - Ответ 6323209

s = b + '#' + a;

#include <iostream>
#include <vector>
#include <string>
#include <conio.h>
using namespace std;
 
string a;
string b;
string s;
int n = 1000000;
 
vector<int>prefix(string s)
{
    s = b + '#' + a;
 
    int n = (int) s.length();
    vector<int> pi (n);
    for (int i=1; i<n; ++i) {
        int j = pi[i-1];
        while (j > 0 && s[i] != s[j])
            j = pi[j-1];
        if (s[i] == s[j]) 
            ++j;
        pi[i] = j;
    }
    return pi;
}
 
int main()
{
    int k, l;
    vector<int>pi(n);
 
    getline(cin,a);
    getline(cin,b);
 
    k = a.length();
    l = b.length();
 
    for(int i = l;i < l+k+1;i++){
        if(pi[i] == l){
            cout << "YES" << endl;
            return 0;
        }
    }
    cout << "NO" << endl;
 
    getch();
 
    return 0;
}