【PTA数据结构 | C语言版】字符串插入操作

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

题目

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

输入格式：
输入先后给出主串 s 和待插入的字符串 t，每个非空字符串占一行，不超过 1000 个字符，以回车结束（回车不算在字符串内）。第三行给出插入的位序 pos（注意位序从 1 开始，保证不超过 s 的长度加 1）。

输出格式：
在一行中输出将 t 插入 s 的第 pos 个字符的位置后的结果字符串。
如果插入后的字符串长度超过了 1000 个字符，则不要插入，而是在一行中输出 错误：插入将导致字符串长度超限。，并且在第二行输出原始主串 s。

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

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

代码

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

#define MAX_LENGTH 1000

int main() {
    char s[MAX_LENGTH + 1], t[MAX_LENGTH + 1];
    int pos;

    // 读取主串 s
    fgets(s, MAX_LENGTH + 1, stdin);
    // 移除换行符
    if (s[strlen(s) - 1] == '\n') {
        s[strlen(s) - 1] = '\0';
    }

    // 读取待插入的字符串 t
    fgets(t, MAX_LENGTH + 1, stdin);
    // 移除换行符
    if (t[strlen(t) - 1] == '\n') {
        t[strlen(t) - 1] = '\0';
    }

    // 读取插入位置 pos
    scanf("%d", &pos);

    // 计算插入后的总长度
    int new_length = strlen(s) + strlen(t);

    if (new_length > MAX_LENGTH) {
        printf("错误：插入将导致字符串长度超限。\n");
        printf("%s\n", s);
    } else {
        // 创建临时数组存储结果
        char result[MAX_LENGTH + 1];
        
        // 复制 s 的前半部分（到 pos-1 位置）
        strncpy(result, s, pos - 1);
        result[pos - 1] = '\0';
        
        // 追加 t
        strcat(result, t);
        
        // 追加 s 的后半部分（从 pos 位置开始）
        strcat(result, s + pos - 1);
        
        printf("%s\n", result);
    }

    return 0;
}