Cod sursa(job #2433677)

Utilizator ArkhamKnightyMarco Vraja ArkhamKnighty Data 28 iunie 2019 16:01:06
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.52 kb
#include <cstring>

#include <fstream>

#define maxn 2000005

std :: ifstream cin("strmatch.in");
std :: ofstream cout("strmatch.out");

void computeLPSArray(char* pat, int M, int* lps);

int rasp[maxn], nr;

void KMPSearch(char* pat, char* txt)
{
    int M = strlen(pat);
    int N = strlen(txt);

    int lps[M];


    computeLPSArray(pat, M, lps);

    int i = 0;
    int j = 0;
    while (i < N)
    {
        if (pat[j] == txt[i])
        {
            j++;
            i++;
        }

        if (j == M)
        {
            rasp[++nr] = i - j ;
            j = lps[j - 1];
        }
        else if (i < N && pat[j] != txt[i])
        {

            if (j != 0)
                j = lps[j - 1];
            else
                i = i + 1;
        }
    }
}

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++;
            }
        }
    }
}

char pat[maxn], txt[maxn];
int main()
{
    cin >> pat >> txt;
    KMPSearch(pat, txt);
    cout << nr << '\n';
    (nr > 1000) ? nr = 1000 : nr = nr;
    for(int i = 1; i <= nr; i++)
        cout << rasp[i] << ' ';
    return 0;
}