Cod sursa(job #1790179)

Utilizator TimitocArdelean Andrei Timotei Timitoc Data 27 octombrie 2016 21:07:35
Problema Iv Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.21 kb
#include <iostream>
#include <cstdio>
#include <cstring>
#define MAXN 550
#define MOD 3210121

using namespace std;

char a[MAXN], b[MAXN];
int n, m, din[2][MAXN][MAXN], rez;

void add(int &to, int who)
{
    to = (to+who) % MOD;
}

void solve()
{
    int ci = 1;
    din[0][0][0] = 1;
    for (int i = 0; i <= n; i++) {
        ci ^= 1;
        for (int j = 0; i+j <= n; j++) {
            for (int st = 0; st <= m; st++) {
                int dr = i+st-j;
                if (dr < 0 || dr > m || din[ci][j][st] == 0) continue;
                if (i+j == n && st + dr == m) add(rez, din[ci][j][st]);
                if (a[i+1] == a[n-j]) add(din[!ci][j+1][st], din[ci][j][st]);
                if (a[i+1] == b[n-dr]) add(din[!ci][j][st], din[ci][j][st]);
                if (b[st+1] == a[n-j]) add(din[ci][j+1][st+1], din[ci][j][st]);
                if (b[st+1] == b[n-dr]) add(din[ci][j][st+1], din[ci][j][st]);
                din[ci][j][st] = 0;
            }
        }
    }
}

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

    gets(a+1);
    gets(b+1);
    n = strlen(a+1);
    m = strlen(b+1);
    solve();
    printf("%d", rez);
    return 0;
}