Cod sursa(job #632846)

Utilizator marcelcodreaCodrea Marcel marcelcodrea Data 12 noiembrie 2011 13:54:12
Problema Potrivirea sirurilor Scor 14
Compilator cpp Status done
Runda Arhiva educationala Marime 1.49 kb
#include<cstdio>
#include<string.h>
#define N 2000005
#define MOD1 100007
#define MOD2 100021
#define BASE 72
int contor;
char a[N];
//int sir[256];
int result[1005];
char b[N];
int ok;
int t, t2;
int NA, NB;
int powb, powc;
int str, str2;
int minim(int a, int b) {
    if(a < b) return a;
    return b;
}
int checkeq(int i) {
    ok = 1;
    for(int j = i; j < i + NA; j++)
       if(b[j] != a[j - i]) ok = 0;
         if(ok == 1) {
            contor++;
            if (contor <= 1000)
                result[contor] = i;
        }
}
int main() {
    freopen("strmatch.in","r",stdin);
    freopen("strmatch.out","w",stdout);
    scanf("%s %s",&a, &b);
    NA = strlen(a);
    NB = strlen(b);
    powb = 1;
    powc = 1;
    for(int i = 0; i < NA; i++) {
        powb = (powb * BASE) % MOD1;
        powc = (powc * BASE) % MOD2;
        t = (t * BASE + a[i]) % MOD1;
        t2 = (t2 * BASE + a[i]) % MOD2;
        str = (str * BASE  + b[i]) % MOD1;
        str2 = (str2 * BASE  + b[i]) % MOD2;
    }
    if((str == t) && (str2 == t2))
     checkeq(0);
    for(int i = 1; i <= NB - NA + 1; i++) {
        str = ((str - (powb * b[i-1]) % MOD1 + MOD1) * BASE + b[i + NA - 1]) % MOD1;
        str2 = ((str2 - (powc * b[i-1]) % MOD2 + MOD2) * BASE + b[i + NA - 1]) % MOD2;
        if((str == t) && (str2 == t2))
            checkeq(i);
    }
    printf("%d\n",contor);
    for(int i = 1; i <= minim(contor, 1000); i++)
     printf("%d ", result[i]);
    return 0;
}