如何在C中编写动态数量的行并将它们赋值给数组

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

int main() {
    int num_rows;
    printf("请输入行数：");
    scanf("%d", &num_rows);

    // 动态分配内存
    int** array = (int**)malloc(num_rows * sizeof(int*));
    if (array == NULL) {
        printf("内存分配失败\n");
        return 1;
    }

    // 为每一行分配内存
    for (int i = 0; i < num_rows; i++) {
        array[i] = (int*)malloc((i + 1) * sizeof(int));
        if (array[i] == NULL) {
            printf("内存分配失败\n");
            return 1;
        }
    }

    // 给每个元素赋值
    for (int i = 0; i < num_rows; i++) {
        for (int j = 0; j <= i; j++) {
            array[i][j] = i + j;
        }
    }

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

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

    return 0;
}