求对称子列表的总和 - Python

对称子列表指的是一个列表中所有对称的子列表，比如[1,2,2,1]的对称子列表有[1],[2],[2],[1],[1,2,2,1],[2,2],[1,2],[2,1]。现在我们需要写一个函数，求一个列表中所有对称子列表的总和。

思路

我们可以遍历列表中的所有子列表，判断其是否是对称的，如果是则将其加入结果中。如何判断一个子列表是否对称呢？我们只需要判断它的正序和反序是否相同即可。

代码实现

def is_palindrome(nums):
    return nums == nums[::-1]

def symmetric_sublists(nums):
    res = []
    for i in range(len(nums)):
        for j in range(i+1, len(nums)+1):
            sublist = nums[i:j]
            if is_palindrome(sublist):
                res.append(sublist)
    return res

def sum_symmetric_sublists(nums):
    sublists = symmetric_sublists(nums)
    return sum(map(sum, sublists))

上面的代码中，我们先定义了一个辅助函数is_palindrome用来判断一个子列表是否是对称的。接着，我们定义了symmetric_sublists函数来寻找所有对称的子列表，最后定义了sum_symmetric_sublists函数来计算所有对称子列表的总和。

使用示例

>>> nums = [1,2,2,1]
>>> sum_symmetric_sublists(nums)
16

>>> nums = [1,2,3,4,5]
>>> sum_symmetric_sublists(nums)
1