Cod sursa(job #1642708)

Utilizator radu_cebotariRadu Cebotari radu_cebotari Data 9 martie 2016 15:47:43
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.11 kb
#include<iostream>
#include<stack>
#include<deque>
#include<queue>
#include<string>
#include<vector>
#include<algorithm>
#include<fstream>
using namespace std;
ifstream in("strmatch.in");
ofstream out("strmatch.out");
const int NMAX = 2000005;
string A,B;
int P[NMAX],sol[1005];

void prefix()
{

    int k = 0;
    P[1] = 0;
    for(int i = 2 ; i < A.size() ; ++i){

        while(k > 0 && A[k+1] != A[i])
            k = P[k];
        if(A[k+1] == A[i])
            ++k;
        P[i] = k;
    }
}

int main()
{

    in>>A>>B;
    A.push_back(' ');
    for(int i = A.size() - 1 ; i >= 0 ; --i)
        A[i+1] = A[i];
    A[0] = ' ';
    prefix();
    int k = 0,n = 0;
    for(int i = 0 ; i < B.size() ; ++i){
        while(k > 0 && A[k+1] != B[i])
            k = P[k];
        if(A[k+1] == B[i])
            ++k;
        if(k == A.size()-1){
            ++n;
            if(n <= 1000)
                sol[n] = i - A.size()+2;
            k = P[A.size()-1];
        }
    }
    out<<n<<"\n";
    for(int i = 1 ; i <= min(n,1000) ; ++i)
        out<<sol[i]<<" ";
    return 0;
}