Cod sursa(job #2647571)

Utilizator tryharderulbrebenel mihnea stefan tryharderul Data 5 septembrie 2020 12:05:52
Problema Prefix Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <iostream>
#include <stdio.h>
#include <string.h>

#define NMAX  2000003

using namespace std;

int n,m,i,k,nr,nr2;
int urm[NMAX];
int v[1003];

char t[NMAX];
char p[NMAX];

void make_prefix(){
    int i,k=-1;

    for(i=1;i<m;i++){
        while(k>0 && p[k+1]!=p[i])
            k=urm[k];

        if(p[k+1]==p[i])k++;
        urm[i]=k;

    }
}

int main()
{
    freopen("strmatch.in","r",stdin);
    freopen("strmatch.out","w",stdout);

    scanf("%s\n",p);
    scanf("%s",t);

    n=strlen(t);
    m=strlen(p);

    make_prefix();

    /*for(i=0;i<m;i++)
        printf("%d ",urm[i]);*/

    k=-1;
    urm[0]=-1;
    for(i=0;i<n;i++){

        while(k>0 && p[k+1]!=t[i])
            k=urm[k];


        if(p[k+1]==t[i])
            k++;

        if(k==m-1){
            nr2++;
            if(nr2<=1000)v[nr2]=i-m+1;
            k=urm[k];

        }

    }

    printf("%d\n",nr2);

   for(i=1;i<=min(nr2,1000);i++)
        printf("%d ",v[i]);

    return 0;
}