Cod sursa(job #2060718)

Utilizator doNotRecallRasa Andrei doNotRecall Data 8 noiembrie 2017 17:23:12
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
#include<fstream>
#include<vector>
using namespace std;
string a, b;
vector<int> sol;
int poz[2000005];
string x;
bool bo;
void a_precalc()
{
    int i = 0;
    int j = 1;
    for(; j < a.size(); )
    {
        if(a[i] == a[j])
        {
            poz[j] = i + 1;
            i++;
            j++;
        }else
        {
            while(i != 0 && a[i] != a[j])
            {
                i = poz[i - 1];
            }
            poz[j] = i;
            j++;
        }
    }
}

void str_match()
{
    int i = 0;
    int j = 0;
    while(j < b.size())
    {
        while(i != 0 && a[i] != b[j])
        {
            i = poz[i - 1];
        }
        if(a[i] == b[j])
        {
            i++;
            j++;
            if(i == a.size())
            {
                if(sol.size() < 1000)
                {
                    sol.push_back(j - i);
                }
                i = poz[i - 1];
            }
        }else
        {
            j++;
        }
    }
}

ifstream in("strmatch.in");
ofstream out("strmatch.out");
int main()
{
    in >> a;
    in >> b;
    a_precalc();
    str_match();
    out << sol.size() << "\n";
    for(int i = 0; i < sol.size(); i++)
    {
        out << sol[i] << " ";
    }
    return 0;
}