如何用Pythonic方式抽象多列表嵌套循环的重叠字符串匹配逻辑

需要找出所有满足以下条件的5位字符串：

• 前三位属于列表l0；
• 第2至4位属于列表l1；
• 第3至5位属于列表l2。

示例输入：

l0=["123","567","451"]
l1=["234","239","881"]
l2=["348","551","399"]

预期输出：['12348', '12399']

已实现判断字符串重叠的函数：

def is_successor(a:str,b:str)->bool:
    """检查字符串a和b是否重叠（a的后两位等于b的前两位）"""
    return a[1:]==b[:2]

当前使用嵌套循环实现，但列表数量增多时可读性差；为减少无效检查，已采用从较长列表（如l2）到较短列表（如l0）的反向循环逻辑，现需抽象这种for→is_successor()的重复逻辑，寻求Pythonic的实现方式。

大规模输入示例：

primes = [2, 3, 5, 7, 11, 13, 17]
lsts=[
    [
        str(j).zfill(3) 
        for j in range(12,988) 
        if not j%prime
    ] 
    for prime in primes
]

核心思路：迭代式拼接 + 反向链式过滤 #

利用functools.reduce抽象多列表的链式匹配逻辑，同时保留从长列表到短列表的反向优化，减少无效匹配次数。

代码实现 #

from functools import reduce

def is_successor(a: str, b: str) -> bool:
    """检查字符串a和b是否重叠（a的后两位等于b的前两位）"""
    return a[1:] == b[:2]

def merge_sequences(prev_results, curr_list):
    """将当前列表元素与上一步结果匹配，生成新的拼接结果"""
    new_results = []
    for curr_item in curr_list:
        for prev_item in prev_results:
            if is_successor(curr_item, prev_item):
                # 拼接逻辑：取curr_item的第一位 + prev_item（重叠部分已匹配，无需重复）
                merged = curr_item[0] + prev_item
                new_results.append(merged)
    return new_results

def find_valid_strings(lists):
    reversed_lists = reversed(lists)
    # 初始结果为最后一个列表的元素
    results = next(reversed_lists)
    # 依次向前匹配每一个列表
    for lst in reversed_lists:
        results = merge_sequences(lst, results)
        # 提前终止：无匹配结果时直接返回空
        if not results:
            break
    return results

# 测试小规模示例
l0=["123","567","451"]
l1=["234","239","881"]
l2=["348","551","399"]
print(find_valid_strings([l0, l1, l2]))  # 输出: ['12348', '12399']

# 处理大规模输入示例
primes = [2, 3, 5, 7, 11, 13, 17]
lsts=[
    [
        str(j).zfill(3) 
        for j in range(12,988) 
        if not j%prime
    ] 
    for prime in primes
]
large_results = find_valid_strings(lsts)
print(large_results[:5])  # 打印前5个结果

优化说明 #

• 反向处理：从最后一个列表开始匹配，快速过滤无效元素，减少后续循环次数
• 逻辑抽象：merge_sequences统一处理每一步的匹配拼接，新增列表时只需添加到输入参数，无需修改循环结构
• 提前终止：某一步无匹配结果时直接返回，避免无效计算

内容的提问来源于stack exchange，提问作者DuesserBaest