计算所有子数组的相等元素的无序对

问题描述：给定一个数组，计算所有子数组中相等元素的无序对数量。

例如，给定数组[1, 2, 3, 1, 2, 1]，所有子数组中相等元素的无序对有：(1, 1), (2, 2)和(1, 1)。

解决方法：使用哈希表（HashMap）和前缀和（prefix sum）。

具体步骤：

1. 创建一个HashMap来存储出现过的前缀和和对应的数量。初始化时，往HashMap中放入(0, 1)，表示前缀和为0的数量为1。

2. 遍历数组，记录当前的前缀和sum，并在HashMap中查找之前是否出现过这个前缀和。如果出现过，则表示这个区间中有相等的元素，将这个前缀和的数量加入答案，即ans += count。

3. 将前缀和sum的数量加1。

代码实现：

def count_equal_pairs(nums):
    count = collections.defaultdict(int)
    count[0] = 1
    ans = 0
    sum = 0
    for num in nums:
        sum += num
        if sum in count:
            ans += count[sum]
        count[sum] += 1
    return ans

时间复杂度为O(n)，空间复杂度为O(n)，其中n为数组长度。

总结：

本问题可以用哈希表来解决，主要是通过前缀和来判断是否有相等元素。使用哈希表可以提高效率，同时也可以减少代码的复杂性。在进行存储数据时可以选择使用collections.defaultdict来避免KeyError的出现，也可以选择先存储前缀和为0的情况。