在C编程语言中，如何在字符串中需要的位置插入数字

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

char* insertDigitAtIndex(const char* str, int digit, int index) {
    size_t len = strlen(str);
    if (index < 0 || index > len) {
        return NULL; // 索引超出范围
    }

    // 分配足够的内存来存储新字符串（原始字符串长度 + 1个数字 + 结尾的空字符）
    char* newStr = (char*)malloc(len + 2);
    if (!newStr) {
        return NULL; // 内存分配失败
    }

    // 复制原始字符串的前半部分
    strncpy(newStr, str, index);
    // 插入数字
    newStr[index] = digit + '0';
    // 复制原始字符串的后半部分
    strcpy(newStr + index + 1, str + index);

    return newStr;
}

int main() {
    const char* originalStr = "HelloWorld";
    int digit = 5;
    int index = 5;

    char* newStr = insertDigitAtIndex(originalStr, digit, index);
    if (newStr) {
        printf("Original string: %s\n", originalStr);
        printf("New string: %s\n", newStr);
        free(newStr); // 记得释放内存
    } else {
        printf("Failed to insert digit or invalid index.\n");
    }

    return 0;
}