2016-06-19 65 views
0

的迭代总和说我有一个列表,列表:使用更好的切片发现亚

A = range(1, 6) = [1, 2, 3, 4, 5] 

B,最终的结果,是列表的列表。鉴于ij,你会如何制作一个迭代和的列表,其中索引i是一边而另一边是j

B[j] = sum(A[j:i+1] or A[i:j+1])取决于ji是否较大。为索引0和2

实例:

B[0] = [1, 1+2, 1+2+3, 1+2+3+4, 1+2+3+4+5] 
    = [1, 3, 6, 10, 15] 
B[2] = [1+2+3, 2+3, 3, 3+4, 3+4+5] 
    = [6, 5, 3, 7, 12] 

======

当前代码(作品)是二for环路,非常强力。我认为应该有一种方法来使用reduce

A = range(1,6) 
n = len(A) 
B = [] 
for j in xrange(n): 
    b = [] 
    for i in xrange(n): 
     if j <= i: 
      b.append(sum(A[j:i+1])) 
     else: 
      b.append(sum(A[i:j+1])) 
    B.append(b) 

# print 
for b in B: 
    print b 

小背景:可能我的解决方案的一部分,以项目欧拉82

回答

1

你最终会重新计算的金额很多倍。相反,创建它们一次,看看他们为b每个元素:

A = range(1,6) 
n = len(A) 
mapping = {} 
for i in xrange(n): 
    for j in xrange(i,n): 
     mapping[i,j] = sum(A[i:j+1]) 

B = [] 
for j in xrange(n): 
    b = [] 
    for i in xrange(n): 
     if j <= i: 
      b.append(mapping[j,i]) 
     else: 
      b.append(mapping[i,j]) 
    B.append(b) 

,你可以无需检查j<=i如果你只是让两个[i,j][j,i]映射工作:

mapping = {} 
A = range(1,6) 
n = len(A) 
for i in xrange(n): 
    for j in xrange(i,n): 
     mapping[i,j] = sum(A[i:j+1]) 
     mapping[j,i] = mapping[i,j] #for both ways 

B = [[mapping[i,j] for i in xrange(n)] for j in xrange(n)] 

虽然请注意,这意味着每个B[x][y]都将直接协调到mapping[x,y],因此您可能只想使用它自己的映射。