Cod sursa(job #2871348)

Utilizator starduststardust stardust Data 14 martie 2022 14:35:17
Problema Potrivirea sirurilor Scor 10
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.53 kb
#include <fstream>
#include <vector>
#include <string>
#include <algorithm>
using namespace std;

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

vector<int> make_prefix(string pattern)

{
    vector<int> prefix(pattern.length(), 0);
    int j = 0;
    int i = 1;

    while (i < pattern.length())
    {
        if (pattern[i] == pattern[j])
        {
            prefix[i] = prefix[j] + 1;
            i++;
            j++;
        }
        else
        {
            if (j == 0)
            {
                prefix[i] = 0;
                i++;
            }
            else
            {
                j = prefix[i - 1];
            }
        }
    }

    return prefix;
}

vector<int> match(string &text, string &pattern)
{
    int n = text.length();
    int m = pattern.length();
    vector<int> result;
    vector<int> prefix = make_prefix(pattern);

    int i = 0;
    int j = 0;

    while (i < n)
    {
        if (text[i] == pattern[j])
        {
            i++;
            j++;
        }
        if (j >= m)
        {
            result.push_back(i - m);
            j = prefix[j - 1];
        }
        else if (i < n && text[i] != pattern[j])
        {
            if (j == 0)
                i++;
            else
                j = prefix[j - 1];
        }
    }

    return result;
}

int main()
{
    string s, pattern;
    getline(in, pattern);
    getline(in, s);

    vector<int> r = match(s, pattern);
    out << r.size() << endl;

    for (int i = 0; i < min(1000, (int)r.size()); i++)

        out << r[i] << " ";

    return 0;
}