Cod sursa(job #1952287)

Utilizator matystroiaStroia Matei matystroia Data 4 aprilie 2017 00:31:09
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.84 kb
#include <fstream>
#include <cstring>
#include <vector>
using namespace std;

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

const int N = 2e6 + 1;

int n, m, pi[N], j;
char a[N], b[N];
vector<int> ans;

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

    j = 0;
    for(int i = 1; i < n; ++i)
    {
        while(j > 0 && a[i] != a[j])
            j = pi[j-1];
        if(a[i] == a[j])
            j++;
        pi[i] = j;
    }

    j = 0;
    for(int i = 0; i < m; ++i)
    {
        while(j > 0 && a[j] != b[i])
            j = pi[j-1];
        if(a[j] == b[i])
            j++;
        if(j == n)
            ans.push_back(i - n + 1);
    }

    fout << ans.size() << '\n';
    for(int i = 0; i < min(1000, (int)ans.size()); ++i)
        fout << ans[i] << " ";

    return 0;
}