Cod sursa(job #1172351)

Utilizator fromzerotoheroFROM ZERO fromzerotohero Data 17 aprilie 2014 12:58:42
Problema Potrivirea sirurilor Scor 2
Compilator cpp Status done
Runda Arhiva educationala Marime 1.16 kb
#include <iostream>
#include <fstream>
#include <cstring>
#include <vector>
using namespace std;

#define nmax 2000005

int k, i, j, n, m;
int pi[nmax],sol[nmax];
// Pi[i] = lungimea celui mai lung prefix din P care este si sufix in P

char P[nmax], T[nmax];

void pp(){
    
    pi[1] = 0;
    
    k = 0;
    
    for (i=2; i<=m; i++){ // cautam cel mai lung prefix din P care este si sufix in sirul P[1..k]
        while (k>0 && P[k+1] != P[i])
            k = P[k];
        if (P[k+1] == P[i]) k++;
        pi[i] = k;
    }
    
}

void kmp(){
    
    for (i=1, k=0; i<=n; i++){
        
        while (k>0 && P[k+1] != T[i]) k = pi[k];
        if (P[k+1] == T[i]) k++;
        
        if (k == m){
            sol[++sol[0]] = i-m;
            k = pi[k];
        }
        
    }
    
}

void tiparire(){

    ofstream out("strmatch.out");
    
    out << min(1000, k) << "\n";
    
    for (i=1; i<=min(1000, k); i++)
        out << sol[i] << " ";
    
    out << "\n";
    
}

int main(){
    
    ifstream in("strmatch.in");
    
    in >> (P+1);
    in >> (T+1);
    
    m = strlen(P+1);
    n = strlen(T+1);
    
    pp();
    kmp();
    
    tiparire();
    
    return 0;
}