Function Name:
odd_even_diag
Parameters:
a 2D list (list of lists)
Returns:
list of lists
Description:
Given a 2-dimensional matrix (list of lists) with any size (n*n), modify it according to the following rules:
- Find the sum of the main diagonal.
- If the sum is an odd number, change all the values of the given matrix (except the main diagonal) to 0.
- If the sum is an even number, change all the values of the given matrix (except the main diagonal) to 1.
Return the resulting matrix.
Example 1:
If argument is:
[[1, 2], [4, 3]]
odd_even_diag should return:
[[1, 1], [1, 3]]
because the sum 1 + 3 is even.
Example 2:
If argument is:
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
odd_even_diag should return:
[[1, 0, 0], [0, 5, 0], [0, 0, 9]]
because the sum 1 + 5 + 9 is odd.