C语言二维数组填充对角线的数字从1到n

#include <stdio.h>
#include <stdlib.h>

void fillDiagonal(int n) {
    int **array = (int **)malloc(n * sizeof(int *));
    for (int i = 0; i < n; i++) {
        array[i] = (int *)malloc(n * sizeof(int));
        for (int j = 0; j < n; j++) {
            array[i][j] = 0; // 初始化为0
        }
    }

    for (int i = 0; i < n; i++) {
        array[i][i] = i + 1; // 填充对角线元素
    }

    // 打印数组
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            printf("%3d ", array[i][j]);
        }
        printf("\n");
    }

    // 释放内存
    for (int i = 0; i < n; i++) {
        free(array[i]);
    }
    free(array);
}

int main() {
    int n = 5; // 示例大小为5
    fillDiagonal(n);
    return 0;
}