Получить квадратную матрицу порядка n - C (СИ)

#include <stdio.h>
#include <stdlib.h>
#define N 9
int main()
{
    int A[N][N] = {{0}}, i, j;
    for(i = 0; i < N; i++){
        A[i][i] = 2;
        if(i < N - 1)
            A[i][i + 1] = 1;
        if(i > 0)
            A[i][i - 1] = 1;
    }
    for(i = 0; i < N; i++, putchar('\n'))
        for(j = 0; j < N; j++)
            printf("%3d", A[i][j]);
    return 0;
}