如何统计逗号分隔数字字符串中，子串和等于指定值的出现次数

需求说明 #

统计逗号分隔的数字字符串中，连续子串对应数字之和等于给定目标值的总数量，不允许使用任何外部库，仅通过基础for循环实现。

测试用例说明 #

• 输入字符串：int_seq='3,0,4,0,3,1,0,1,0,1,0,0,5,0,4,2'
• 目标和值：subtotal = 9
• 预期输出：7，匹配的连续子串共7组，如下：

原串：3,0,4,0,3,1,0,1,0,1,0,0,5,0,4,2
匹配1：_0,4,0,3,1,0,1,0_____________
匹配2：_0,4,0,3,1,0,1_______________
匹配3：___4,0,3,1,0,1,0_____________
匹配4：___4,0,3,1,0,1_______________
匹配5：___________________0,0,5,0,4_
匹配6：_____________________0,5,0,4_
匹配7：_______________________5,0,4_

实现代码（纯原生Python，无外部依赖） #

# 入参定义
int_seq = '3,0,4,0,3,1,0,1,0,1,0,0,5,0,4,2'
subtotal = 9

# 1. 处理输入字符串为整数列表
num_list = [int(num) for num in int_seq.split(',')]
count = 0
list_length = len(num_list)

# 2. 双层for循环遍历所有连续子串
# 外层循环：子串的起始索引
for start_idx in range(list_length):
    current_total = 0
    # 内层循环：从起始索引开始逐步扩展子串的结束索引，同步累加和
    for end_idx in range(start_idx, list_length):
        current_total += num_list[end_idx]
        if current_total == subtotal:
            count += 1

print(count)

逻辑说明 #

• 外层for循环遍历所有可能的连续子串起始位置
• 内层for循环从当前起始位置向后扩展子串长度，边扩展边计算当前子串的和，避免重复计算
• 每次判断当前和等于目标值时，直接累加计数
• 因为存在数值为0的元素，扩展时和可能先等于目标值、加0后仍然等于目标值，因此不需要在和超过目标值时提前终止内层循环

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