Ideas2IT interview question

Matrix Diagonal addition