Combination Lock

def checknum(digit,c,index,n):
    return not(abs(digit-c[index]) > 2 and abs(abs(digit-c[index])-n) > 2)

def check(digits,c1,c2,n):
    for c in [c1,c2]:
        res = True
        for i in range(len(digits)):
            if not checknum(digits[i],c,i,n):
                res = False
        if res == True:
            return True
    return False

def bruteForce(n,c1,c2):
    res = 0
    for i in range(1,n+1):
        if checknum(i,c1,0,n) or checknum(i,c2,0,n):
            for j in range(1,n+1):
                if checknum(j,c1,1,n) or checknum(j,c2,1,n):
                    for k in range(1,n+1):
                        if check([i,j,k],c1,c2,n):
                            res += 1
                            #print(i,j,k,end='    ')
    return res

def main():
    with open('combo.in','r') as fIn:
        r = fIn.readlines()
        n = int(r[0])
        c1 = list(map(int,r[1].split()))
        c2 = list(map(int,r[2].split()))
    res = bruteForce(n,c1,c2)
    print(res)
    with open('combo.out','w') as fOut:
        fOut.write(str(res)+'\n')

if __name__ == '__main__':
    main()