Cod sursa(job #3231654)

Utilizator Ruxandra009Ruxandra Vasilescu Ruxandra009 Data 27 mai 2024 15:37:54
Problema Potrivirea sirurilor Scor 14
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.53 kb
#include <fstream>
#include <cstring>
#include <vector>

using namespace std;

ifstream f("strmatch.in");
ofstream g("strmatch.out");

const int mod1 = 666013;
const int mod2 = 10005;
const int nmaxx = 2000005;

int n, m, sumok1, sumok2, sumi1, sumi2;
char a[nmaxx], b[nmaxx];
vector<int> v;

int power(int a, int b, int mod)
{
    int sol = 1;
    while(b)
    {
        if(b % 2 == 0)
        {
            a = (a * a) % mod;
            b /= 2;
        }

        else
        {
            sol = (sol * a) % mod;
            b --;
        }
    }

    return sol;
}

int main()
{
    f >> a >> b;
    n = strlen(a);
    m = strlen(b);

    for(int i = 0; i < n; i ++)
    {
        sumok1 = (sumok1 + a[i] * power(127, n - i - 1, mod1)) % mod1;
        sumok2 = (sumok2 + a[i] * power(127, n - i - 1, mod2)) % mod2;
    }

    for(int i = 0; i < n; i ++)
    {
        sumi1 = (sumi1 + b[i] * power(127, n - i - 1, mod1)) % mod1;
        sumi2 = (sumi2 + b[i] * power(127, n - i - 1, mod2)) % mod2;
    }

    if(sumi1 == sumok1 && sumi2 == sumok2)
        v.push_back(0);

    for(int i = n; i < m; i ++)
    {
        sumi1 = ((sumi1 - (b[i - n] * power(127, n - 1, mod1)) % mod1 + mod1) % mod1) * 127 + b[i];
        sumi1 %= mod1;

        sumi2 = ((sumi2 - (b[i - n] * power(127, n - 1, mod2)) % mod2 + mod2) % mod2) * 127 + b[i];
        sumi2 %= mod2;

        if(sumi1 == sumok1 && sumi2 == sumok2)
            v.push_back(i - n + 1);
    }

    g << v.size() << '\n';
    for(int i = 0; i < v.size(); i ++)
        g << v[i] << " ";
    return 0;
}