Cod sursa(job #1503179)

Utilizator visanrVisan Radu visanr Data 15 octombrie 2015 17:58:12
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.22 kb
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;

const int NMAX = 2000010;

int N, M, Z[NMAX];
char A[NMAX], B[NMAX];
int Cnt;
vector<int> Pos;

void BuildZ()
{
    int C = 1, R = 1;
    for(int i = 2; i <= N; ++ i)
    {
        if(i <= R) Z[i] = min(Z[i - C + 1], R - i + 1);
        else Z[i] = 0;

        while(i + Z[i] <= N && A[i + Z[i]] == A[Z[i] + 1]) Z[i] ++;

        if(i + Z[i] - 1 > R) C = i, R = i + Z[i] - 1;
    }
}

void FindMatches()
{
    int C = 0, R = 0;

    for(int i = 1; i <= M; ++ i)
    {
        int LCP = 0;
        if(R >= i) LCP = min(Z[i - C + 1], R - i + 1);

        while(i + LCP <= M && LCP <= N && A[LCP + 1] == B[i + LCP]) LCP ++;

        if(LCP == N)
        {
            Cnt ++;
            if(Cnt <= 1000) Pos.push_back(i - 1);
        }

        if(i + LCP - 1 > R) C = i, R = i + LCP - 1;
    }
}

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

    gets(A + 1); N = strlen(A + 1);
    gets(B + 1); M = strlen(B + 1);

    BuildZ();
    FindMatches();

    printf("%i\n", Cnt);
    for(int i = 0; i < Pos.size(); ++ i) printf("%i ", Pos[i]);
}