Pagini recente » Cod sursa (job #2015061) | Istoria paginii utilizator/teodoramusatoiu | Monitorul de evaluare | Istoria paginii runda/contest_2 | Cod sursa (job #1406779)
#include <iostream>
#include <fstream>
#include <cstring>
#include <cstdlib>
using namespace std;
#define N_MAX 2000001
char A[N_MAX], B[N_MAX];
typedef struct t_matches {
int len, pos[N_MAX];
} t_matches;
t_matches m;
void computeLPSArray(char *pat, int M, int *lps)
{
int len = 0; // lenght of the previous longest prefix suffix
int i;
lps[0] = 0; // lps[0] is always 0
i = 1;
// the loop calculates lps[i] for i = 1 to M-1
while (i < M)
{
if (pat[i] == pat[len])
{
len++;
lps[i] = len;
i++;
}
else // (pat[i] != pat[len])
{
if (len != 0)
{
// This is tricky. Consider the example AAACAAAA and i = 7.
len = lps[len-1];
// Also, note that we do not increment i here
}
else // if (len == 0)
{
lps[i] = 0;
i++;
}
}
}
}
void KMP(char *pat, char *txt, t_matches *matches)
{
int M = strlen(pat);
int N = strlen(txt);
matches->len = 0;
// create lps[] that will hold the longest prefix suffix values for pattern
int *lps = (int *)malloc(sizeof(int)*M);
int j = 0; // index for pat[]
// Preprocess the pattern (calculate lps[] array)
computeLPSArray(pat, M, lps);
int i = 0; // index for txt[]
while (i < N)
{
if (pat[j] == txt[i])
{
j++;
i++;
}
if (j == M)
{
matches->pos[matches->len++] = i - j;
j = lps[j-1];
}
// mismatch after j matches
else if (i < N && pat[j] != txt[i])
{
// Do not match lps[0..lps[j-1]] characters,
// they will match anyway
if (j != 0)
j = lps[j-1];
else
i = i+1;
}
}
free(lps); // to avoid memory leak
}
int main()
{
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
fin >> A >> B;
KMP(A, B, &m);
fout << m.len << "\n";
for (int i = 0; i < m.len; ++i) {
fout << m.pos[i] << " ";
}
fout << "\n";
return 0;
}