python for循环 实现子数组之和为0

在Python中，你可以使用`for`循环结合列表推导式和条件判断来找到所有元素的子数组，其和恰好等于0。这种方法通常被称为“零和子数组”问题。以下是一个简单的例子：

def zero_sum_subarray(arr):
    # 利用字典存储每个数以及它的负数形式出现的位置
    count = {0: 1}  # 初始化，键值对 (0, 1) 表示有一个数组元素为0
    result = []  # 存储满足条件的子数组

    # 遍历输入数组
    for i in range(len(arr)):
        # 对当前元素，更新它在负数形式下的计数
        arr[i] = -arr[i] if arr[i] < 0 else arr[i]
        if arr[i] in count:
            # 如果正负相加等于0，找到一对相反数
            # 将它们对应的索引添加到结果子数组，并移除已找到的一对
            result.extend([[j, i+1] for j in range(i+count[arr[i]])])
            del count[arr[i]]
        
        # 更新当前元素的计数
        if arr[i] not in count:
            count[arr[i]] = 0
        count[arr[i]] += 1
    
    return result

# 测试示例
arr = [15, -3, 4, -2, 11, 9, -4, 6, 8]
print(zero_sum_subarray(arr))  # 输出所有子数组的起始和结束位置

在这个函数中，我们检查是否能找到一个元素及其负数出现在数组中，使得它们两个的和为0。如果找到了，就记录下它们的索引。这个过程会一直持续到遍历完整个数组。