Квадратная матрица
Пусть у нас есть двумерная квадратная матрица размерами $N \times N$ тогда условие принадлежности главной её диагонали это просто:
$ i = j$
где $i$ и $j$ -- индексы строки и столбца (в том или ином порядке).
а вот условие принадлежности побочной диагонали подразумевает, что если мы движемся от элемента к элементу построчно, то должны выполняться одновременно два условия (запишем их, используя знак логического умножения):
$ (j = N + 1 - i) \wedge (i = N + 1 - j)$