Python 如何检查某个相邻子数组的和是否等于N?

Python 如何检查某个相邻子数组的和是否等于N?,python,python-3.x,Python,Python 3.x,假设我有一个这样的序列列表 我想删除所有序列,其中它的总和=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

假设我有一个这样的序列列表

我想删除所有序列,其中它的总和=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

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

  • 列表中的元素总计为N。然后您可以简单地测试:

    if sum(myList) == N:
        # do fancy stuff
    
  • 列表中的元素总和不小于N。在这种情况下,可能有一个子序列总和为N。要找到它,让我们定义两个指针,
    l
    r
    。它们的名称代表
    ,将定义子序列的边界。然后,解决方案如下:

    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
    
    r=1
    l=0
    r 4:
    l+=1
    其他:
    #做些花哨的事情,然后退出循环
    打破
    
    它的工作原理如下。首先初始化<代码> L>代码>代码> R>代码>,以便考虑只包含第一个元素的子序列> MyList。然后,对子序列的元素求和,如果和小于N,则通过将
    1
    添加到
    r
    来放大子序列。如果大于N,则通过将
    1
    添加到
    l
    来限制子序列


  • 注意:感谢eozd


    上述算法仅在列表元素非负时有效。

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

    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)]
    

    像(1,1,3)这样的小序列中的数字是非负的吗?还有,这些“小”序列的最大长度是多少?是的,它们保证为负,最大长度是N-1。你的意思是它们保证为非负?如果是这样的话,Neb给出了一个非常有效的线性时间解。@Thegreatfoo so,我认为在N有多大以及元素的负性条件是什么的问题上也值得一提。*非负,我的错误。在这种情况下,N的范围是2。第二种算法仅当序列中的元素为非负时才起作用。我认为这值得一提。你是对的,我假设它们都是非负的,因为OPIf最大长度为
    N-1
    ,OP可能需要它来推广到任何
    N