Cod sursa(job #1997123)

Utilizator MateiAruxandeiMateiStefan MateiAruxandei Data 3 iulie 2017 14:28:23
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 2.11 kb
#include <fstream>
#include <stdio.h>
#include <string.h>

// d is the number of characters in input alphabet
#define d 256

using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
/* pat -> pattern
    txt -> text
    q -> A prime number
*/

char txt[2000005], pat[2000005];
int rez[1005];
int nr;

void search(char pat[], char txt[], int q)
{
    int M = strlen(pat);
    int N = strlen(txt);
    int i, j;
    int p = 0; // hash value for pattern
    int t = 0; // hash value for txt
    int h = 1;

    // The value of h would be "pow(d, M-1)%q"
    for (i = 0; i < M - 1; i++)
        h = (h * d) % q;

    // Calculate the hash value of pattern and first
    // window of text
    for (i = 0; i < M; i++)
    {
        p = (d * p + pat[i]) % q;
        t = (d * t + txt[i]) % q;
    }

    // Slide the pattern over text one by one
    for (i = 0; i <= N - M; i++)
    {

        // Check the hash values of current window of text
        // and pattern. If the hash values match then only
        // check for characters on by one
        if ( p == t )
        {
            /* Check for characters one by one */
            for (j = 0; j < M; j++)
            {
                if (txt[i + j] != pat[j])
                    break;
            }

            // if p == t and pat[0...M-1] = txt[i, i+1, ...i+M-1]
            if (j == M && nr <= 1000)
                rez[++nr] = i;
        }

        // Calculate hash value for next window of text: Remove
        // leading digit, add trailing digit
        if (i < N - M )
        {
            t = (d * (t - txt[i] * h) + txt[i + M]) % q;

            // We might get negative value of t, converting it
            // to positive
            if (t < 0)
            t = (t + q);
        }
    }
}

/* Driver program to test above function */
int main()
{
    fin.getline(pat, 2000005);
    fin.getline(txt, 2000005);

    int q = 101;

    search(pat, txt, q);

    fout << nr << '\n';
    for(int i = 1; i <= nr; ++i)
		fout << rez[i] << ' ';
    fout << '\n';
    return 0;
}