Pagini recente » Cod sursa (job #289997) | preONI 2008 - Clasament Runda 3, Clasele 5-8 | Cod sursa (job #2368052) | Cod sursa (job #215532) | Cod sursa (job #1294891)
#include<fstream>
#include<string.h>
using namespace std;
int prefix[2000000];
char s[2000000];
char subs[2000000];
int nr=0;
ofstream out("strmatch.out");
ifstream in("strmatch.in");
void KMP_Prefix(char s[],int prefix[])
{
int L=strlen(s);
prefix[0]=0;
int a=0;
for(int b=1;b<L;b++)
{
if(s[b]==s[a])
{
prefix[b]=prefix[b-1]+1;
++a;
}
else if(s[b]!=s[a])
{
a=0;
prefix[b]=prefix[a];
}
}
}
void KMP(char s[],char subs[])
{
int M=strlen(subs);
int N=strlen(s);
KMP_Prefix(subs,prefix);
int i=0,j=0;
for(int i=0;i<N;i++)
{
if(s[i]==subs[j])
{
j++;
if(j==M)
{
nr++;
if(nr<=1000)
out<<i-M+1<<' ';
j=prefix[j-1];
}
}
else if(j > 0 && s[i]!=subs[j])
{
j=prefix[j-1];
if(j>0)
i--;
}
}
}
int main()
{
in>>subs;
in>>s;
out<<' '<<'\n';
KMP(s,subs);
out.seekp(0);
out<<nr;
in.close();
out.close();
return 0;
}