n сформировать матрицу A[n][n], в которой элементы диагонали равны номеру строки, а все остальные элементы - нулевые. Написать программу на С++

#include <stdio.h>

int main()

{

   unsigned n;

   /* организовать ввод квадратной матрицы размера nxn из целых чисел; */

   printf("Enter matrix dimension: ");

   scanf("%u", &n);

   int matrix[n][n];

   for (unsigned i = 0; i < n; i++)

   {

       for (unsigned j = 0; j < n; j++)

       {

           if (i == j)

           {

               matrix[i][i] = i + 1;

           }

           else

           {

               matrix[i][j] = 0;

           }

       }

   }

   for (unsigned i = 0; i < n; i++)

   {

       for (unsigned j = 0; j < n; j++)

       {

           printf("%d ", matrix[i][j]);

       }

       printf("");

   }

}