Cod sursa(job #2484302)

Utilizator DanutAldeaDanut Aldea DanutAldea Data 30 octombrie 2019 21:45:12
Problema Iv Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.6 kb
#include <fstream>
#include <cstring>
#define mod 3210121
using namespace std;

ifstream fin("iv.in");
ofstream fout("iv.out");

int n,N,m,i,j,l,d[2][510][510],sol,t;
char a[510],b[510];

int main(){
    fin>>a+1;
    n=strlen(a+1);
    fin>>b+1;
    m=strlen(b+1);

    if(a[1]==a[n])
        d[1][1][1]=1;
    if(a[1]==b[m])
        d[1][1][0]=1;
    if(b[1]==a[n])
        d[1][0][1]=1;
    if(b[1]==b[m])
        d[1][0][0]=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[1-t][i-1][j-1];
                    d[t][i][j]%=mod;
                }

                if(a[i]==b[m-l+j+1] && i>0 && m-l+j+1<=m){
                    d[t][i][j]+=d[1-t][i-1][j];
                    d[t][i][j]%mod;
                }

                if(b[l-i]==a[n-j+1] && l-i>0 && n-j+1<=n){
                    d[t][i][j]+=d[1-t][i][j-1];
                    d[t][i][j]%=mod;
                }

                if(b[l-i]==b[m-l+j+1] && l-i>0 && m-l+j+1<=m){
                    d[t][i][j]+=d[1-t][i][j];
                    d[t][i][j]%=mod;
                }
            }
        }

        t=1-t;
    }
    t=1-t;

    if((n+m)%2)
        for(i=0;i<n;i++){
            sol+=d[t][i][n-i];
            sol%=mod;
            sol+=d[t][i][n-i-1];
            sol%=mod;
        }
    else
        for(i=0;i<=n;i++){
            sol+=d[t][i][n-i];
            sol%=mod;
        }

    fout<<sol;

    return 0;
}