in python

Develop a program that receives an array of integers and displays a list of the main diagonal elements (as a list). The program should read the data the number of lines and columns and the data of the matrix and should show the main diagonal elements as a list. In case the matrix is not a square matrix, it should show "the matrix is not square".

Example:

3 4

Output

the matrix is not square

example 3:

3 3 1 2 3 4 5 6 7 8 9 

Output

[1, 5, 9]

output