def sum_of_diagonals(two_d_list):
"""[[1,3,2,4],[8,7,5,6],[15,13,12,11],[9,-8,-10,-6]]"""
main_diag = []
vice_diag = []
for i in range(len(two_d_list)):
for j in range(len(two_d_list)):
if j == i:
main_diag.append(two_d_list[i][j])
if j + i == len(two_d_list) - 1:
vice_diag.append(two_d_list[i][j])
return sum(main_diag), sum(vice_diag)
def input_accept_multi_lines(end_ch='/'):
str = ''
for line in iter(input, end_ch):
str += line + '\n'
return str
input_num_list = list(map(int,input_accept_multi_lines().split()[:16]))
row_len = int(pow(len(input_num_list),1/2))
two_d_list = [[input_num_list[j] for j in range(i*row_len, (i+1)*row_len)] for i in range(row_len)]
main,vice = sum_of_diagonals(two_d_list)
print(main+vice)
相关文章：