Envío 4823

Código

N = 0
a = []
matrix = []
sum_total = 0
magic_sum = 0
answer = ''

N = int(input())

for index in range(N):
    a = []
    for num in input().split(' '):
        x = int(num)
        sum_total +=x
        a.append(x)
    matrix.append(a)

magic_sum =sum_total/N

def rows_check(matrix, magic_sum, answer, N):
    for row in range(N):
        if sum(matrix[row]) != magic_sum:
            answer = 'No'
            break
        else:
            answer = 'Yes'
    return(answer)


def column_check(matrix, magic_sum, answer, N):
    for column in range(N):
        if sum([rows[column] for rows in matrix]) != magic_sum:
            answer = 'No'
            break
        else:
            answer = 'Yes'
    return(answer)


def top_diagonal_check(matrix, magic_sum, answer, N):
    top_diagonal = 0
    for row_index in range(N):
        for column_index in range(N):
            if row_index == column_index:
                top_diagonal += matrix[row_index][column_index]
            else:
                continue

    if top_diagonal != magic_sum:
        answer = 'No'
    else:
        answer = 'Yes'

    return(answer)

def lower_diagonal_check(matrix, magic_sum, answer, N):
    lower_diagonal = 0
    diagonal_index = N-1
    for row_index in range(N):
        for column_index in range(N):
            if column_index == diagonal_index:
                lower_diagonal += matrix[row_index][column_index]
                diagonal_index += -1
            else:
                continue

    if lower_diagonal != magic_sum:
        answer = 'No'
    else:
        answer = 'Yes'

    return(answer)

answer = rows_check(matrix, magic_sum, answer, N)

if answer == 'Yes':
    answer = column_check(matrix, magic_sum, answer, N)

if answer == 'Yes':
    answer = top_diagonal_check(matrix, magic_sum, answer, N)

if answer == 'Yes':
    answer = lower_diagonal_check(matrix, magic_sum, answer, N)


print(answer)