Cod sursa(job #3151101)

Utilizator andutu0Lungu Andrei andutu0 Data 19 septembrie 2023 19:08:15
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.38 kb
#include <iostream>
#include <fstream>
#include <algorithm>
#include <cmath>
#include <vector>
#include <utility>
#include <queue>
#include <stack>
#include <set>
#include <string.h>
using namespace std;

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

typedef long long ll;

char pat[2000000],txt[2000000];
int lps[2000000];
int n,m;
int ans[1001],cnt=0;

void prefix()
{
    int len = 0;
    int i = 1;
    lps[0]=0;
    while(i<m)
    {
        if(pat[i]==pat[len])
        {
            lps[i]=len+1;
            len++;
            i++;
        }
        else{
            if(len!=0) len=lps[len-1];
                else{
                    lps[i]=0;
                    i++;
                }
        }
    }
}

void kmp()
{
    n = strlen(txt);
    m = strlen(pat);
    prefix();
    int i=0,j=0;
    while(i<n)
    {
        if(txt[i]==pat[j])
        {
            i++;
            j++;
        }
        else
        {
            if(j!=0) j=lps[j-1];
            else i++;
        }
        if(j==m)
        {
            cnt++;
            if(cnt <= 1001)
            {
                ans[cnt] = i-j;
            }
            j=lps[j-1];
        }
    }
}

int main()
{
    fin>>pat>>txt;
    kmp();
    fout<<cnt<<'\n';
    for(int i=1 ; i<=min(cnt,1000); i++)
        fout<<ans[i]<<' ';
    return 0;
}