Cod sursa(job #1344398)

Utilizator tudorv96Tudor Varan tudorv96 Data 16 februarie 2015 18:14:36
Problema Potrivirea sirurilor Scor 52
Compilator cpp Status done
Runda Arhiva educationala Marime 0.88 kb
#include <fstream>
#include <cstring>
#include <vector>
using namespace std;

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

const int N = 2e6 + 5;

char a[N], b[N];

int U[N], k = 0, m, n;
vector <int> v;

#define min(x,y) ((x)<(y)?(x):(y))

int main() {
    fin >> a + 1 >> b + 1;
    m = strlen(a+1);
    n = strlen(b+1);
    for (int i = 2; i <= m; ++i) {
        while (k && a[k + 1] != a[i])
            k = U[k];
        if (a[k + 1] == a[i])
            k++;
        U[i] = k;
    }
    for (int i = 1; i <= n; ++i) {
        while (k && a[k+1] != b[i])
            k = U[k];
        if (a[k + 1] == b[i])
            k++;
        if (k == m) {
            v.push_back (i - k);
            k = U[k];
        }
    }
    fout << v.size() << "\n";
    for (unsigned i = 0; i < min(v.size(), 1000); ++i)
        fout << v[i] << " ";
}