Codewars_exercises/find_the_unknown_digit.py at main · iliyan-pigeon/Codewars_exercises · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
def solve_runes(runes):
    unknown_digit = -1
    runes_copy = runes

    for i in range(10):
        if str(i) not in runes_copy:
            while "?" in runes_copy:
                index = runes_copy.index("?")
                runes_copy = runes_copy[:index] + str(i) + runes_copy[index+1:]

            first_number = []
            operator = ""
            second_number = []
            numbers, result = runes_copy.split("=")
            numbers = list(numbers)
            result = list(result)

            for j in range(len(numbers)-1):
                if numbers[j] == "-" and j == 0:
                    continue
                if numbers[j] == "-" or numbers[j] == "+" or numbers[j] == "*":
                    operator = numbers[j]
                    first_number = numbers[:j]
                    second_number = numbers[j+1:]
                    break

            if first_number[0] == "0" and len(first_number) > 1 or \
                    second_number[0] == "0" and len(second_number) > 1:
                runes_copy = runes
                continue

            if result[0] == "0" and len(result) > 1:
                runes_copy = runes
                continue

            if operator == "+":
                if int("".join(first_number)) + int("".join(second_number)) == int("".join(result)):
                    unknown_digit = i
                    break
            elif operator == "-":
                if int("".join(first_number)) - int("".join(second_number)) == int("".join(result)):
                    unknown_digit = i
                    break
            elif operator == "*":
                if int("".join(first_number)) * int("".join(second_number)) == int("".join(result)):
                    unknown_digit = i
                    break
            runes_copy = runes

    return unknown_digit


print(solve_runes("?*11=??"))