【PTA数据结构 | C语言版】字符串插入操作（不限长）

本专栏持续输出数据结构题目集，欢迎订阅。

题目

请编写程序，将给定字符串 t 插入到另一个给定字符串 s 的第 pos 个字符的位置。

输入格式：
输入先后给出主串 s 和待插入的字符串 t，每个非空字符串占一行，长度无固定上限，以回车结束（回车不算在字符串内）。第三行给出插入的位序 pos，是 int 范围内的任意整数（注意正常的位序从 1 开始）。

输出格式：
在一行中输出将 t 插入 s 的第 pos 个字符的位置后的结果字符串。
如果插入的位序 pos 不合法，则不要插入，而是在一行中输出 错误：指定插入位置不存在。，并且在第二行输出原始主串 s。

输入样例：
This is a test.
simple-
11

输出样例：
This is a simple-test.

代码

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

#define MAX_LENGTH 1000000

int main() {
    char s[MAX_LENGTH] = {0};
    char t[MAX_LENGTH] = {0};
    char result[2 * MAX_LENGTH] = {0};
    
    // 读取主串s
    if (fgets(s, MAX_LENGTH, stdin) == NULL) {
        return 1;
    }
    // 移除换行符
    size_t len_s = strlen(s);
    if (len_s > 0 && s[len_s - 1] == '\n') {
        s[len_s - 1] = '\0';
        len_s--;
    }
    
    // 读取待插入字符串t
    if (fgets(t, MAX_LENGTH, stdin) == NULL) {
        return 1;
    }
    // 移除换行符
    size_t len_t = strlen(t);
    if (len_t > 0 && t[len_t - 1] == '\n') {
        t[len_t - 1] = '\0';
        len_t--;
    }
    
    // 读取插入位置
    int pos;
    if (scanf("%d", &pos) != 1) {
        return 1;
    }
    
    // 检查插入位置是否合法
    if (pos < 1 || pos > len_s + 1) {
        printf("错误：指定插入位置不存在。\n");
        printf("%s\n", s);
    } else {
        // 复制s的前半部分
        memcpy(result, s, pos - 1);
        // 插入t
        memcpy(result + pos - 1, t, len_t);
        // 复制s的后半部分
        memcpy(result + pos - 1 + len_t, s + pos - 1, len_s - (pos - 1));
        
        // 输出结果
        printf("%s\n", result);
    }
    
    return 0;
}