如何生成字符串全元素的所有分组组合？代码问题及优化求助

你需要生成的是字符串的所有连续子串拆分组合——要求使用全部字符、保持原有顺序，本质是枚举字符串中所有可能的分割点组合（每个字符之间可选择拆分或不拆分）。

现有代码的问题 #

你的代码存在两个关键缺陷：

• 数据结构错误：用字典存储结果不符合需求，且辅助函数仅做单次拆分，未递归处理剩余字符串的所有可能拆分路径
• 逻辑漏洞：生成了包含空字符串的无效组合，且未将拆分结果正确拼接为完整的分组列表

解决方案1：递归实现 #

递归思路直观易懂：每次从字符串开头截取1到k个字符作为第一个分组，然后对剩余字符串递归生成所有拆分方式，再将第一个分组与每个递归结果拼接，直到字符串为空时返回终止条件。

def get_all_splits(s):
    if not s:
        return [[]]
    splits = []
    # 遍历所有可能的首段长度
    for i in range(1, len(s) + 1):
        first_segment = s[:i]
        # 递归获取剩余部分的所有拆分
        rest_splits = get_all_splits(s[i:])
        # 拼接首段与剩余拆分结果
        for split in rest_splits:
            splits.append([first_segment] + split)
    return splits

# 测试示例
for group in get_all_splits("12345"):
    print(group)

解决方案2：利用itertools枚举分割点 #

对于长度为n的字符串，有n-1个潜在分割点（比如"12345"的分割点在字符1-2、2-3、3-4、4-5之间）。每个分割点有"拆分"或"不拆分"两种选择，共2^(n-1)种组合。我们可以用itertools.product生成所有选择，再按选择拆分字符串。

import itertools

def get_all_splits_iter(s):
    n = len(s)
    if n == 0:
        return [[]]
    splits = []
    # 生成所有分割点的选择：0=不拆分，1=拆分
    for split_choices in itertools.product([0, 1], repeat=n-1):
        current_group = []
        start_idx = 0
        for idx, choice in enumerate(split_choices):
            if choice == 1:
                current_group.append(s[start_idx:idx+1])
                start_idx = idx + 1
        # 添加最后一段字符
        current_group.append(s[start_idx:])
        splits.append(current_group)
    return splits

# 测试示例
for group in get_all_splits_iter("12345"):
    print(group)

输出示例（以"123"为例） #

['123']
['1', '23']
['12', '3']
['1', '2', '3']

内容的提问来源于stack exchange，提问作者Hitesh Reddy Mumma Reddy