Cod sursa(job #2245073)

Utilizator GeoeyMexicanuBadita George GeoeyMexicanu Data 24 septembrie 2018 18:03:31
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include <bits/stdc++.h>

using namespace std;

vector<int> q;
char pat[2000010],txt[2000010];
int M,lps[2000010],N;
void computeLPSArray();
void KMPSearch()
{
    M=strlen(pat);
    N=strlen(txt);
    computeLPSArray();
    int i=0;
    int j=0;
    while(i<N)
    {
        if(pat[j]==txt[i])
        {
            i++;
            j++;
        }
        if(j==M)
        {
            q.push_back((i-j));
            j=lps[j-1];
        }
        else
            if(i<N && pat[j]!=txt[i])
            {
                if(j!=0)
                    j=lps[j-1];
                else
                    i=i+1;
            }
    }
}
void computeLPSArray()
{
    int len=0;
    lps[0]=0;
    int i=1;
    while(i<M)
    {
        if(pat[i]==pat[len])
        {
            len++;
            lps[i]=len;
            i++;
        }
        else
        {
            if(len!=0)
            {
                len=lps[len-1];
            }
            else
            {
                lps[i]=0;
                i++;
            }
        }
    }
}
ifstream f("strmatch.in");
ofstream g("strmatch.out");
int main()
{
    f>>pat;
    f>>txt;
    KMPSearch();
    g<<q.size()<<'\n';
    for(int i=0;i<q.size();i++)
        g<<q[i]<<' ';
    return 0;
}