Python | 重复字符串直到 K

在Python中，我们可以使用循环和字符串操作来重复一个字符串，直到其长度达到K。这是一个常见的编程任务，尤其在竞赛编程中。

以下是一个简单的Python函数，可以重复给定字符串S，直到其长度达到K。如果S本身的长度已经大于等于K，则它将只是返回已有的字符串。

def repeat_string(S: str, K: int) -> str:
    n = len(S)
    if n >= K:
        return S[:K]
    else:
        return (S * (K // n)) + S[:K % n]

让我们看看这个函数是如何工作的。

如何工作

首先，我们计算字符串S的长度，并检查是否已经大于等于K。如果是的话，我们只返回S的前K个字符。

如果S的长度小于K，我们需要重复它，直到达到长度K。为此，我们使用S的长度n和K的商，得到一个整数q。我们将S重复q次，然后将S的前K % n个字符附加到结果字符串的末尾。这样，我们就得到了一个长度为K的字符串，其中S已经被重复了若干次。

让我们看看这个函数是否可靠。考虑以下几个测试用例：

assert repeat_string('abc', 10) == 'abcabcabca'
assert repeat_string('hello', 3) == 'hel'
assert repeat_string('', 5) == ''

第一个测试用例中，原始字符串'abc'被重复三倍，直到其长度达到10，所以我们得到'abcabcabca'。第二个测试用例中，原始字符串'hello'无法达到长度3，所以我们只返回'hel'。第三个测试用例中，我们没有字符串可重复，所以结果为空字符串。

这个函数看起来可以处理各种情况，让我们看看它的时间复杂度和空间复杂度是多少。

时间复杂度和空间复杂度

这个函数的时间复杂度和空间复杂度都是O(K)，其中K是目标长度。在最坏的情况下，我们需要将S重复K // n次，这需要O(K)次运算。此外，我们需要存储最多K个字符的结果字符串，这需要O(K)的额外空间。因此，总时间复杂度和空间复杂度都是O(K)。

结论

在Python中，重复给定字符串直到其长度达到目标长度是一项常见的任务。我们可以使用循环和字符串操作来轻松地实现它。当我们需要一个函数来处理这种情况时，可以使用这个简单而有效的Python函数。