Question

所以说我有一个这样的列表序列。

我要删除所有总和= N和/或具有和的子数组且总和= N的所有序列。

例如，如果N = 4，则（1,1,2）无效，因为其总数为4。（1,1,3）也是无效的，因为（1,3）也是4。（1,3,1）出于同样的原因也无效。

lst = [ 
    (1,1,1), (1,1,2), (1,1,3), 
    (1,2,1), (1,2,2), (1,2,3), 
    (1,3,1), (1,3,2), (1,3,3), 
    (2,1,1), (2,1,2), (2,1,3), 
    (2,2,1), (2,2,2), (2,2,3), 
    (2,3,1), (2,3,2), (2,3,3), 
    (3,1,1), (3,1,2), (3,1,3), 
    (3,2,1), (3,2,2), (3,2,3), 
    (3,3,1), (3,3,2), (3,3,3) 
]

有哪些方法可以做到这一点？

我目前正在尝试查看是否能够删除其总数不一定是N的倍数但不是其连续子数组但我不成功的序列

   for elements in list(product(range(1,n), repeat=n-1)):
        lst.append(elements)
    for val in lst:
        if np.cumsum(val).any() %n != 0:
            lst2.append(val) # append value to a filtered list

Answer 1

您可以将问题分为两个子问题：

列表中的元素总计N。然后您可以简单地测试：
```
if sum(myList) == N:
    # do fancy stuff
```
列表中的元素不总和为N。在这种情况下，可能会有一个子序列，总和为N。要找到它，让我们定义两个指针，{ {1}}和l。它们的名称代表r和left，它们将定义子序列的边界。然后，解决方案如下：
```
right
```
它的工作原理如下。首先，初始化r = 1 l = 0 while r <= len(myList): sum_ = sum(myList[l:r]) if sum_ < 4: r += 1 elif sum_ > 4: l += 1 else: # do fancy stuff and exit from the loop break和l，以便考虑仅由r的第一个元素组成的子序列。然后，对子序列的元素求和，如果总和小于N，则通过在myList上加上1来放大子序列。如果它大于N，则可以通过在r上加上1来限制子序列。

感谢eozd ：

仅当列表的元素为非负数时，以上算法才有效。

Answer 2

您可以使用itertools.combinations生成切片索引的所有组合以测试子序列的总和：

from itertools import combinations
[t for t in lst if not any(sum(t[l:h+1]) == 4 for l, h in combinations(range(len(t)), 2))]

这将返回：

[(1, 1, 1), (1, 2, 3), (2, 3, 2), (2, 3, 3), (3, 2, 1), (3, 2, 3), (3, 3, 2), (3, 3, 3)]

如何检查某个连续子数组的和是否等于N？

2 个答案: