Cod sursa(job #1290043)

Utilizator tudi98Cozma Tudor tudi98 Data 10 decembrie 2014 19:13:23
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.06 kb
#include <fstream>
#include <vector>
using namespace std;

vector<int> Sol;

int pi[2000005];

void compute_pi(string A,int n)
{
    pi[1] = 0;
    int k = 0;
    for(int i = 1; i < n; i++)
    {
        while(k > 0 && A[k] != A[i])
            k = pi[k+1];
        if(A[i] == A[k])
            k++;
        pi[i+1] = k;
    }
}

int main()
{
    ifstream fin("strmatch.in");
    ofstream fout("strmatch.out");

    string A,B;
    fin >> A >> B;
    int n = A.size();
    int m = B.size();

    if(n > m)
    {
        fout << "0";
        return 0;
    }

    compute_pi(A,n);

    int Nr = 0;
    int k = 0;
    for(int i = 0; i < m; i++)
    {
        while(k > 0 && A[k] != B[i])
            k = pi[k];
        if(A[k] == B[i])
        {
            k++;
            if(k == n)
            {
                Nr++;
                if(Nr <= 1000) Sol.push_back(i-n+1);
            }
        }
        else k = pi[k];
    }

    fout << Nr << "\n";
    for(int i = 0; i < Sol.size(); i++)
        fout << Sol[i] << " ";

}