Python 如何检查某个相邻子数组的和是否等于N?
假设我有一个这样的序列列表 我想删除所有序列,其中它的总和=N和/或它有一个连续的子数组,总和=N 例如,如果N=4,则(1,1,2)无效,因为其总数为4。(1,1,3)也无效,因为(1,3)也是4。(1,3,1)也因同样的原因无效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
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
您可以将问题分为两个子问题:
if sum(myList) == N:
# do fancy stuff
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