ツクダオリジナル 算数塾の回答を導出するプログラムの作成

column1 = ["1","2","5","3","4"]#各面の値を記載
column2 = ["-","-","*","+","*"]
column3 = ["2","5","4","3","1"]
column4 = ["*","+","-","-","/"]
column5 = ["4","5","2","3","1"]
answer = ["4","2","3","1","5"]
ans_count = 0

def judge_add(i,j,k,l,m,n):#[0,1,2,3,4]となっているため5を超えた場合0に戻す処理
    if i > 4:
        return_i = i-5
    else:
        return_i = i
    if j > 4:
        return_j = j-5
    else:
        return_j = j
    if k > 4:
        return_k = k-5
    else:
        return_k = k
    if l > 4:
        return_l = l-5
    else:
        return_l = l

    if m > 4:
        return_m = m-5
    else:
        return_m = m

    if n > 4:
        return_n = n-5
    else:
        return_n = n

    return return_i,return_j,return_k,return_l,return_m,return_n
def add_one(i,j,k,l,m,n):#他の面が妥当か検証するためにiなどの値に1追加する関数
    i+=1
    j+=1
    k+=1
    l+=1
    m+=1
    n+=1
    return i,j,k,l,m,n


for i in range(5):
    for j in range(5):
        for k in range(5):
            for l in range(5):
                for m in range(5):
                    for n in range(5):
                        join_column1 = column1[i]+column2[j]+column3[k]+column4[l]+column5[m]
                        if int(eval(join_column1)) == int(answer[n]):#まず1列が正しいか判定し、正しければ他の列の評価する
                            add_i,add_j,add_k,add_l,add_m,add_n = i,j,k,l,m,n#一旦別の変数で現在のｉなどの値を保持
                            for o in range(5):#他の5面の検証
                                add_i,add_j,add_k,add_l,add_m,add_n = add_one(add_i,add_j,add_k,add_l,add_m,add_n)
                                judge_i,judge_j,judge_k,judge_l,judge_m,judge_n = judge_add(add_i,add_j,add_k,add_l,add_m,add_n)
                                join_column = column1[judge_i]+column2[judge_j]+column3[judge_k]+column4[judge_l]+column5[judge_m]
                                if int(eval(join_column)) == int(answer[judge_n]):
                                    ans_count += 1#検証した結果問題妥当と判断されたらカウンタをインクり
                            if ans_count >= 4:#4面が妥当と判断されたら回答を返答
                                print("YES")
                                print(join_column1+"="+answer[n])
                                ans_count = 0
                            else:
                                ans_count = 0                

YES
1*5-1=4
YES
2+4-4=2
YES
5*3/5=3
YES
3-1*2=1
YES
4-2+3=5