Cod sursa(job #2485561)

Utilizator divianegoescuDivia Negoescu divianegoescu Data 1 noiembrie 2019 19:06:24
Problema Iv Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.26 kb
#include <fstream>
#include <cstring>
#define K 3210121
using namespace std;
ifstream fin("iv.in");
ofstream fout("iv.out");
int n,m,i,j,l,sol,t;
int d[2][510][510];
char a[510],b[510];
int main(){
    fin>>a+1>>b+1;
    n=strlen(a+1);
    m=strlen(b+1);
    if(a[1]==a[n])d[1][1][1]=1;
    if(b[1]==b[m])d[1][0][0]=1;
    if(a[1]==b[m])d[1][1][0]=1;
    if(b[1]==a[n])d[1][0][1]=1;
    for(l=2;l<=(n+m)/2;l++){
        for(i=0;i<=n && i<=l;i++){
            for(j=0;j+i<=n && j<=l;j++){
                d[t][i][j]=0;
                if(a[i]==a[n-j+1] && i>0 && n-j+1<=n)
                    d[t][i][j]=(d[t][i][j]+d[1-t][i-1][j-1])%K;
                if(b[l-i]==b[m-l+j+1] && l-i>0 && m-l+j+1<=m)
                    d[t][i][j]=(d[t][i][j]+d[1-t][i][j])%K;
                if(a[i]==b[m-l+j+1] && i>0 && m-l+j+1<=m)
                    d[t][i][j]=(d[t][i][j]+d[1-t][i-1][j])%K;
                if(b[l-i]==a[n-j+1] && l-i>0 && n-j+1<=n)
                    d[t][i][j]=(d[t][i][j]+d[1-t][i][j-1])%K;
            }
        }
        t=1-t;
    }
    t=1-t;
    if((n+m)%2)
        for(i=0;i<n;i++)
            sol=(sol+d[t][i][n-i]+d[t][i][n-i-1])%K;
    else
        for(i=0;i<=n;i++)
            sol=(sol+d[t][i][n-i])%K;
    fout<<sol;
    return 0;
}