Cod sursa(job #893678)

Utilizator andreea29Iorga Andreea andreea29 Data 26 februarie 2013 17:10:36
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.14 kb
#include<fstream>
#include<cstring>

#define Nmax 2000010

using namespace std;

int n, m, i, p, prefix[Nmax], sol[Nmax];
char a[Nmax], b[Nmax], as[Nmax], bs[Nmax];

int main()
{
    ifstream f("strmatch.in");
    ofstream h("strmatch.out");
    f.getline (as, Nmax);
    f.getline (bs, Nmax);
    n = strlen (as);
    m = strlen (bs);
    for (i = 0; i < n; ++i)
        a[i + 1] = as[i];
    for (i = 0; i < m; ++i)
        b[i + 1] = bs[i];
    f.close();

    p = 0;

    prefix[1] = 0;
    for (i = 2; i <= n; ++i)
    {
        while (p > 0 && a[p + 1] != a[i])
            p = prefix[p];

        if (a[p + 1] == a[i])
            ++p;

        prefix[i] = p;
    }

    p = 0;
    for (i = 1; i <= m; ++i)
    {
        while (p > 0 && a[p + 1] != b[i])
            p = prefix[p];

        if (a[p + 1] == b[i])
            ++p;

        if (p == n)
        {
            ++sol[0];
            sol[sol[0]] = i - n;
            p = prefix[p];
        }
    }

    h << sol[0] << '\n';
    n = min (1000, sol[0]);
    for (i = 1; i <= n; ++i)
        h << sol[i] << " ";

    h.close();
    return 0;
}