Pagini recente » Cod sursa (job #1308661) | Cod sursa (job #2256894) | Cod sursa (job #1245578) | Cod sursa (job #1710882) | Cod sursa (job #1099721)
#include <fstream>
#include <iostream>
#include <vector>
#include <bitset>
#include <string.h>
#include <algorithm>
#include <iomanip>
#include <math.h>
#include <time.h>
#include <stdlib.h>
#include <set>
#include <map>
#include <string>
#include <queue>
#include <deque>
using namespace std;
const char infile[] = "strmatch.in";
const char outfile[] = "strmatch.out";
ifstream fin(infile);
ofstream fout(outfile);
const int MAXN = 2000005;
const int oo = 0x3f3f3f3f;
typedef vector<int> Graph[MAXN];
typedef vector<int> :: iterator It;
const inline int min(const int &a, const int &b) { if( a > b ) return b; return a; }
const inline int max(const int &a, const int &b) { if( a < b ) return b; return a; }
const inline void Get_min(int &a, const int b) { if( a > b ) a = b; }
const inline void Get_max(int &a, const int b) { if( a < b ) a = b; }
/***
* We have to match A in B
*
***/
char A[MAXN], B[MAXN];
int Prefix[MAXN];
int N, M;
inline void buildPrefix(char *a, int *Pi, const int &N) {
int K = 0;
for(int i = 2 ; i <= N ; ++ i) {
while(K > 0 && a[K + 1] != a[i])
K = Pi[K];
if(a[K + 1] == a[i])
++ K;
Pi[i] = K;
}
}
inline vector<int> KMP(char *a, char *b, const int &N, const int &M) {
vector <int> matches;
buildPrefix(a, Prefix, N);
int K = 0;
for(int i = 1 ; i <= M ; ++ i) {
while(K > 0 && a[K + 1] != b[i])
K = Prefix[K];
if(a[K + 1] == b[i])
++ K;
if(K == N)
matches.push_back(i - N);
}
return matches;
}
int main() {
fin.getline(A + 1, MAXN);
fin.getline(B + 1, MAXN);
N = strlen(A + 1);
M = strlen(B + 1);
vector<int> matches = KMP(A, B, N, M);
fout << matches.size() << '\n';
for(int i = 0 ; i < min(matches.size(), 1000 ) ; ++ i)
fout << matches[i] << ' ';
fin.close();
fout.close();
return 0;
}