Вывести главную диагональ матрицы - C (СИ)

Формулировка задачи:

Дано натуральное число n, которое определяет соответственно количество строк и столбцов в некоторой заданной квадратной матрицы целых чисел. Вывести главную диагональ матрицы.

Решение задачи: «Вывести главную диагональ матрицы»

Листинг программы

#include <stdio.h>
 
#define N 3
 
int main() {
    int i, j;
    int a[N][N] = {
        {1,2,3},
        {4,5,6},
        {7,8,9}
    };
    printf("Матрица %dx%d\n", N, N);
    for (i = 0; i < N; ++i) {
        for (j = 0; j < N; ++j) {
            printf("%5d", a[i][j]);
        }
        putchar('\n');
    }
    puts("Элементы главной диагонали");
    for (i = 0; i < N; ++i) {
        printf("%5d", a[i][i]);
    }
    putchar('\n');
    return 0;    
}

Объяснение кода листинга программы

В данном коде:

Объявляется массив a размером NxN для хранения элементов матрицы.
Инициализируются элементы матрицы.
Выводится размерность матрицы.
В цикле два вложенных цикла выводятся все элементы матрицы.
Выводится главная диагональ матрицы.
Возвращается 0, чтобы указать, что программа успешно завершилась.