Cod sursa(job #2564324)

Utilizator Tudor_PascaTudor Pasca Tudor_Pasca Data 1 martie 2020 20:09:52
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.42 kb
#include <iostream>
#include <fstream>
#include <string>

using namespace std;

ifstream in("strmatch.in");
ofstream out("strmatch.out");

string x, y;
int lps[2000010];
int rez[1010];

void buildLps()
{
    int sze = y.size();
    int len = 0, i = 1;

    lps[0] = 0;

    while(i < sze)
    {
        if(y[i] == y[len])
        {
            len++;

            lps[i] = len;

            i++;
        }
        else
        {
            if(len != 0)
                len--;

            else
            {
                len = 0;

                lps[i] = 0;

                i++;
            }
        }
    }
}

void solveKMP()
{
    int n = x.size();
    int m = y.size();

    int i = 0;
    int j = 0;

    int k = 0;

    while(i < n)
    {
        if(x[i] == y[j])
        {
            i++;
            j++;
        }

        if(j == m)
        {
            rez[k] = i - j;

            k++;

            //cout << rez[k - 1] << ' ' << i - j << '\n';

            j = lps[j - 1];
        }

        else if(i < n && x[i] != y[j])
        {
            if(j == 0)
                i++;
            else
                j = lps[j-1];
        }
    }

    k = min(k, 1000);

    out << k << '\n';

    for(int i = 0; i < k; i++)
        out << rez[i] << ' ';
}

int main()
{
    in >> y >> x;

    buildLps();

    solveKMP();

    return 0;
}