arrays_isCryptSol.py

"""
A cryptarithm is a mathematical puzzle for which the goal is to
find the correspondence between letters and digits,
such that the given arithmetic equation consisting of letters holds true when the letters are converted to digits.

You have an array of strings crypt, the cryptarithm, and an an array containing the mapping of letters and digits,
solution. The array crypt will contain three non-empty strings that follow the structure:
[word1, word2, word3], which should be interpreted as the word1 + word2 = word3 cryptarithm.

If crypt, when it is decoded by replacing all of the letters in the cryptarithm with digits using the 
mapping in solution, becomes a valid arithmetic equation containing no numbers with leading zeroes,
the answer is true. If it does not become a valid arithmetic solution, the answer is false.

Note that number 0 doesn't contain leading zeroes (while for example 00 or 0123 do).

Example

For crypt = ["SEND", "MORE", "MONEY"] and

solution = [['O', '0'],
            ['M', '1'],
            ['Y', '2'],
            ['E', '5'],
            ['N', '6'],
            ['D', '7'],
            ['R', '8'],
            ['S', '9']]
the output should be
isCryptSolution(crypt, solution) = true.

When you decrypt "SEND", "MORE", and "MONEY" using the mapping given in crypt, you get 9567 + 1085 = 10652 which is correct and a valid arithmetic equation.

For crypt = ["TEN", "TWO", "ONE"] and

solution = [['O', '1'],
            ['T', '0'],
            ['W', '9'],
            ['E', '5'],
            ['N', '4']]
the output should be
isCryptSolution(crypt, solution) = false.

Even though 054 + 091 = 145, 054 and 091 both contain leading zeroes, meaning that this is not a valid solution.
"""
def isCryptSolution(crypt, solution):
    solution_dict = dict(solution)
    first = [solution_dict[val][0] for val in crypt[0]]
    second = [solution_dict[val][0] for val in crypt[1]]
    result = [solution_dict[val][0] for val in crypt[2]]
    
    if (len(first) > 1 and first[0] == '0') or (len(second) > 1 and second[0] == '0'):
        return False

    added = int(''.join(first)) + int(''.join(second))
    
    if result[0] == '0' and len(result) > 1:
        return False
    if added == int(''.join(result)):
        return True
    return False