Cod sursa(job #1316695)

Utilizator retrogradLucian Bicsi retrograd Data 13 ianuarie 2015 23:54:09
Problema Potrivirea sirurilor Scor 16
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include<fstream>
#include<vector>
#include<cstring>
#define MAXN 2000002
#define SOLSIZE 1000

using namespace std;

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

char STRING[MAXN], WORD[MAXN];
int PI[MAXN+1], wordlen;

#define PI (PI+1)

void process() {
    fin>>WORD[0];
    int i;
    for(i=0; WORD[i++] != '\n';) {
        fin>>noskipws>>WORD[i];
        if(WORD[i] == WORD[PI[i-1]])
            PI[i] = PI[i-1] + 1;
        else PI[i] = 0;
    }
    WORD[i] = 0;
    wordlen = i-1;
}

vector<int> SOL;
int ap;

void search() {
    int stare = 0;
    for(int i=0; STRING[i++] != '\n';) {
        fin>>noskipws>>STRING[i];
        if(STRING[i] == WORD[stare]) {
            stare++;
            if(stare == wordlen) {
                ap++;
                if(SOL.size() <= SOLSIZE)
                    SOL.push_back(i);
                stare = PI[stare-1];
            }
        } else {
            stare = PI[stare];
        }
    }
}

void afis() {
    fout<<ap<<'\n';
    for(vector<int>::iterator it=SOL.begin(); it!=SOL.end(); ++it) {
        fout<<*it - wordlen<<" ";
    }
}

int main() {
    process();
    search();
    afis();
    return 0;
}