главной диагонали

Есть стандартная задача по нахождению суммы элементов главной диагоналиРешаю таким образом:

Вот код :

Есть такой код, знаю не самый лучший, но есть нужно переделать что бы он заменял элементы главной диагонали каждой строки arsum'омСейчас это частично но он заменяет немного не так