Python – 将字符串分成相等的 K 块

在编写Python程序时，经常需要对字符串进行拆分，以方便进行各种操作。本篇文章介绍如何将字符串分成相等的K块。

方法一：使用字符串切片

我们可以通过对字符串进行切片，将字符串分成K块。首先确定每块的长度，然后依次截取字符串的子串，直到截取完为止。具体代码如下：

def divide_str(s: str, k: int):
    n = len(s)
    if n % k != 0:
        return None
    seg_len = n // k
    res = []
    for i in range(k):
        res.append(s[i * seg_len: (i + 1) * seg_len])
    return res

s = "abcdefgh"
k = 4
print(divide_str(s, k)) # ['abcd', 'efgh']

该方法的时间复杂度为O(n)，空间复杂度为O(k)。

方法二：使用正则表达式

我们也可以使用正则表达式对字符串进行分割。具体实现如下：

import re

def divide_str(s: str, k: int):
    n = len(s)
    if n % k != 0:
        return None
    seg_len = n // k
    pattern = f".{{{seg_len}}}"
    res = re.findall(pattern, s)
    return res

s = "abcdefgh"
k = 4
print(divide_str(s, k)) # ['abcd', 'efgh']

该方法同样具有O(n)的时间复杂度和O(k)的空间复杂度。

总结

本文介绍了两种方法将字符串分成相等的K块，其中一种是使用字符串切片，另一种是使用正则表达式。根据具体情况可选取适合的方法。