本专栏持续输出数据结构题目集,欢迎订阅。
文章目录
题目
请编写程序,将给定字符串 t 插入到另一个给定字符串 s 的第 pos 个字符的位置。
输入格式:
输入先后给出主串 s 和待插入的字符串 t,每个非空字符串占一行,长度无固定上限,以回车结束(回车不算在字符串内)。第三行给出插入的位序 pos,是 int 范围内的任意整数(注意正常的位序从 1 开始)。
输出格式:
在一行中输出将 t 插入 s 的第 pos 个字符的位置后的结果字符串。
如果插入的位序 pos 不合法,则不要插入,而是在一行中输出 错误:指定插入位置不存在。,并且在第二行输出原始主串 s。
输入样例:
This is a test.
simple-
11
输出样例:
This is a simple-test.
代码
cpp
#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;
}