The Imitation Game

#include <cstdio>
#include <cstring>
using namespace std;

#define D(x) 

int f[6][30];
int g[6][30];
int state[6];
int cipher[90];
int plain[90];
int len;
int *mark[3];
int mark_num;
bool ok;
int power[] = {1, 26, 26 * 26, 26 * 26 * 26};

int get_next(int a[])
{
    char st[90];
    scanf("%s", st);
    for (int i = 0; st[i]; i++)
    {
        if (st[i] == ‘?‘)
        {
            a[i] = -1;
            mark[mark_num++] = &a[i];
            continue;
        }
        a[i] = st[i] - ‘a‘;
    }
    return strlen(st);
}

void input()
{
    mark_num = 0;
    for (int i = 0; i < 5; i++)
    {
        get_next(f[i]);
    }
    get_next(state);
    len = get_next(plain);
    get_next(cipher);
}

void output()
{
    for (int i = 0; i < len; i++)
    {
        putchar(plain[i] + ‘a‘);
    }
    puts("");
}

int encrypt(int a, int pos)
{
    int temp = f[4][a];
    int cur_state[4];
    int delta[4];

    for (int i = 0; i < 4; i++)
    {
        cur_state[i] = state[i] + pos / power[i];
        cur_state[i] = (cur_state[i] + 26) % 26;
    }
    delta[0] = cur_state[0];
    for (int i = 1; i < 4; i++)
    {
        delta[i] = (cur_state[i] - cur_state[i - 1] + 26) % 26;
    }

    for (int i = 0; i < 4; i++)
    {
        temp = f[i][(temp + delta[i] + 26) % 26];
    }

    for (int i = 2; i >= 0; i--)
    {
        temp = g[i][(temp - delta[i + 1] + 26) % 26];
    }

    temp = g[4][(temp - delta[0] + 26) % 26];
    return temp;
}

bool check()
{
    for (int i = 0; i < 5; i++)
    {
        for (int j = 0; j < 26; j++)
        {
            g[i][f[i][j]] = j;
        }
    }

    for (int i = 0; i < len; i++)
    {
        if (encrypt(plain[i], i) != cipher[i])
        {
            return false;
        }
    }
    return true;
}

void dfs(int step)
{
    if (ok)
    {
        return;
    }
    if (step == mark_num)
    {
        if (check())
        {
            ok = true;
            output();
        }
        return;
    }
    for (int i = 0; i < 26; i++)
    {
        *mark[step] = i;
        dfs(step + 1);
    }
}

void test(char a, int b)
{
    check();
    printf("%c %c\n", a, encrypt(a - ‘a‘, b) + ‘a‘);
}

int main()
{
    int t;
    scanf("%d", &t);
    for (int i = 1; i <= t; i++)
    {
        printf("Scenario #%d:\n", i);
        input();
        ok = false;
        dfs(0);
        D(test(‘a‘, 1));
        puts("");
    }
    return 0;
}