a+b=c 숫자식에서 각 숫자의 한자리 수 치환을 통해 식을 맞게 만드는 방법

a + b = c 일때, 각 숫자 a, b, c 중 하나의 숫자를 바꿔서 식을 맞게 만들 수 있는 갯수를 구하기

바꿀 수 있는 숫자는 1 ~ 9 까지이며, 0은 사용할 수 없다. 

a 를 바꿔서 a 와 b 의 합이 c 가 되도록, b 를 바꿔서 a 와 b 의 합이 c 가 되도록, 혹은 c 를 바꿔서 a 와 b 의 합이 c 가 되도록 고쳐야 한다. 

각 숫자의 자릿수는 상관없이 입력되며, 하나의 숫자에서 한 자릿수만 수정할 수 있다. 예를 들어 1 -> 9 로 바꾸거나, 123 -> 143 으로 바꿀 수 있다. 

 

 

st = '1121+10010=11140'

def solution(st):
    a, temp = st.split('+')
    b, c = temp.split('=')
    inta, intb, intc = int(a), int(b), int(c)
    res = 0
    f = 0
    if inta + intb == intc:
        res = 1
        f = 1
    if f == 0:
        lenA, lenB, lenC = len(a), len(b), len(c)
        def check(size, strNum, num, ans):
            tmpStr = strNum
            for i in range(size):
                strNum = tmpStr
                for j in range(1, 10):
                    temp = list(strNum)
                    temp[i] = str(j)
                    strNum = "".join(temp)
                    if int(strNum) + num == ans:
                        return 1
            return 0
        def checkc(size, strNum, inta, intb):
            tmpStr = strNum
            for i in range(size):
                strNum = tmpStr
                for j in range(1, 10):
                    temp = list(strNum)
                    temp[i] = str(j)
                    strNum = "".join(temp)
                    if inta + intb == int(strNum):
                        return 1
            return 0
        res += check(lenA, a, intb, intc)
        res += check(lenB, b, inta, intc)
        res += checkc(lenC, c, inta, intb)
    print(res)

solution(st)