国际空间研究组织 | ISRO CS 2017 – 5 月 | 问题 26

在这个题目中，给出一个字符串和它的一个子字符串，要求将该子字符串插入到该字符串的指定位置。

输入格式

输入一行包含三个字符串 $s$，$t$ 和 $p$，其中 $s$ 表示原始字符串，$t$ 表示需要插入的子字符串，$p$ 表示需要将子字符串插入的位置。$s$ 和 $t$ 的长度不超过 $100$，$p$ 是一个非负整数。

输出格式

输出将 $t$ 插入到 $s$ 的 $p$ 位置之后得到的新字符串。

解题思路

首先，需要根据 $p$ 将原始字符串切分成两部分，然后将需要插入的子字符串与后半部分拼接，得到新的字符串。具体实现如下：

s = input().strip()
t = input().strip()
p = int(input().strip())

result = s[:p] + t + s[p:]

print(result)

测试样例

输入：

hello world
beautiful
6

输出：

hello beautiful world

时间复杂度

该算法只需要一次字符串拼接操作，时间复杂度为 $O(n)$，其中 $n$ 表示原始字符串的长度。