【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 1000000int main() {char s[MAX_LENGTH] = {0};char t[MAX_LENGTH] = {0};char result[2 * MAX_LENGTH] = {0};// 读取主串sif (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--;}// 读取待插入字符串tif (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);// 插入tmemcpy(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;
}