Cod sursa(job #1374500)

Utilizator CosminRusuCosmin Rusu CosminRusu Data 5 martie 2015 09:38:49
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.9 kb
#include <iostream>
#include <vector>
#include <fstream>
#include <string.h>

using namespace std;

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

const int maxn = 2000005;

int n, m, pi[maxn];
char a[maxn], b[maxn];
vector <int> matches;

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