Cod sursa(job #2680514)

Utilizator SoulSnorterPetre Robert Cristian SoulSnorter Data 3 decembrie 2020 18:07:07
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.66 kb
#include <stdio.h>
#include <vector>
#include <string.h>

using namespace std;

#define NMax 2000001
char A[NMax], B[NMax];

int lps[1024];

void computeLPSArray(char* pat, int M, int lps[])
{
    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++;
            }
        }
    }
}

int main()
{
    freopen("strmatch.in", "rt", stdin);
    freopen("strmatch.out", "wt", stdout);
    scanf("%s %s", A, B);
    int nr = 0;
    vector <int> sir;
    int M = strlen(A);
    int N = strlen(B);
    computeLPSArray(A, M, lps);
    /* afisare array lps
    for (int i = 0; i < M; i++)
    {
        cout << lps[i] << " ";
    }

    cout << '\n';
    */

    if (M > N)
    {
        printf("0\n");
        return 0;
    }

    int j = 0;
    int i = 0;
    while (i < N && nr != 1000)
    {
        if (A[j] == B[i])
        {
            j++;
            i++;
        }
        if (j == M && j != 0)
        {
            j = lps[j - 1];
            sir.push_back(i - M);
            nr++;
        }
        else if (i < N && A[j] != B[i])
        {
            if (j != 0)
                j = lps[j - 1];
            else
                i++;
        }
    }
    printf("%d\n", nr);
    for (int i = 0; i < sir.size(); i++)
    {
        printf("%d ", sir[i]);
    }
    return 0;
}