Cod sursa(job #1648988)

Utilizator VoicencuVoicencu Teodor Octavian Voicencu Data 11 martie 2016 12:12:58
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.04 kb
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <algorithm>
#include <cmath>
#include <string>
#include <iomanip>
#include <stack>
#include <string.h>
#define pb push_back
#define mp make_pair

using namespace std;

ifstream f("strmatch.in");
ofstream g("strmatch.out");

vector<int> sol;

int p[2000001],n,m;
char s[2000001],t[2000001];

void citire()
{
    f>>s;
    n=strlen(s);
    f>>t;
    m=strlen(t);
}

void prefix(char t[])
{
    int k=0;
    for(int i=1;i<n;i++)
    {
        while(k>0 and t[k]!=t[i])k=p[k-1];
        if(t[k]==t[i])k++;
        p[i]=k;
    }
}

void kmp(char t[],char s[])
{
    prefix(s);
    int k=0;
    for(int i=0;i<m;i++)
    {
        while(k>0 and s[k]!=t[i] ) k=p[k-1];
        if(s[k]==t[i])k++;
        if(k==n) sol.push_back(i-n+1);
    }
}

void rezolva()
{
    kmp(t,s);
    g<<sol.size()<<'\n';
    for(int i=0; i<min((int)sol.size(),1000); i++)g<<sol[i]<<' ';
}

int main()
{
    citire();
    rezolva();
    return 0;
}