015. Forming a Magic Square.py

# Problem: https://www.hackerrank.com/challenges/magic-square-forming
# Score: 20


def formingMagicSquare(square):
    allSquares = [
        [[4, 9, 2], [3, 5, 7], [8, 1, 6]],
        [[2, 7, 6], [9, 5, 1], [4, 3, 8]],
        [[6, 1, 8], [7, 5, 3], [2, 9, 4]],
        [[8, 3, 4], [1, 5, 9], [6, 7, 2]],
        [[2, 9, 4], [7, 5, 3], [6, 1, 8]],
        [[6, 7, 2], [1, 5, 9], [8, 3, 4]],
        [[8, 1, 6], [3, 5, 7], [4, 9, 2]],
        [[4, 3, 8], [9, 5, 1], [2, 7, 6]],
        ]
    minCost = 9 * 9
    for magicSquare in allSquares:
        cost = 0
        for i in range(3):
            for j in range(3):
                cost += abs(magicSquare[i][j] - square[i][j])
        if cost < minCost:
            minCost = cost
    return minCost


square = [
    list(map(int, input().split())),
    list(map(int, input().split())),
    list(map(int, input().split()))
    ]
print(formingMagicSquare(square))