[已解决]Python：每日一题 312

zltzlt

今天的题目：

给定一个数组，返回数组中的所有子数组之和。

说明：如果数组为 [2, 4, 1]，则该数组所有的子集是 [[2], [4], [1], [2, 4], [4, 1], [2, 4, 1]]。

示例 1：

输入：[1, 2, 3]
输出：20
解释：(1) + (2) + (3) + (2 + 3) + (1 + 2) + (1 + 2 + 3) = 20

示例 2：

输入：[1, 2]
输出：6
解释：(1) + (2) + (1 + 2) = 6

欢迎大家一起答题！

最佳答案

月排行榜 / 总排行榜

newbison

2020-1-22 18:20:15

上一题不会，这一题我用递归来解决吧。实在看不懂你们的高深的方法。

1. def func312(array):
   
2.     l = len(array)
   
3.     res = 0
   
4.     if l == 1:
   
5.         res = array[0]
   
6.     elif l == 2:
   
7.         res = array[0] + array[1] + (array[0] + array[1])
   
8.     else:
   
9.         res = func312(array[:(l-1)])
   
10.         for i in range(0, l):
    
11.             res += add(array[i:])
    
12.     #递归法解决该问题，每添加一个元素，多出来的数组统统加一遍
    
13.     return res
    
14. 
    
15. def add(array):
    
16.     res = 0
    
17.     for each in array:
    
18.         res += each
    
19.     return res
    
20. 
    
21. 
    
22. print('array [1, 2, 3]的答案是: ', func312([1, 2, 3]))
    
23. print('array [1, 2]的答案是: ', func312([1, 2]))
    
24. print('array [5, 2, 8, 4, 2]的答案是: ', func312([5, 2, 8, 4, 2]))
    

复制代码

跳转到最佳答案楼层

TJBEST

哇，楼主真的太勤奋了@zltzlt  这个重复元素怎么处理？比如[2,2,3,4,3,4]这个您能给示范吗？

阴阳神万物主

迷惑行为。
[2, 4, 1] -> [[2],[1],[4],[2,1],[1,4],[2,4],[2,4,1]]
[1, 2, 3] -> [[1],[2],[3],[1,2],[1,3],[2,3],[1,2,3]]

zltzlt

阴阳神万物主 发表于 2020-1-22 14:13
迷惑行为。
[2, 4, 1] -> [[2],[1],[4],[2,1],[1,4],[2,4],[2,4,1]]
[1, 2, 3] -> [[1],[2],[3],[1,2],[1 ...

子数组必须是连续的

TJBEST

阴阳神万物主 发表于 2020-1-22 14:13
迷惑行为。
[2, 4, 1] -> [[2],[1],[4],[2,1],[1,4],[2,4],[2,4,1]]
[1, 2, 3] -> [[1],[2],[3],[1,2],[1 ...

[3,4,3,4]这样的数组 是不是存在两个[3,4]啊？还是算一个

阴阳神万物主

zltzlt 发表于 2020-1-22 14:18
子数组必须是连续的

哦

zltzlt

TJBEST 发表于 2020-1-22 14:09
哇，楼主真的太勤奋了@zltzlt  这个重复元素怎么处理？比如[2,2,3,4,3,4]这个您能给示范吗？

和普通数组一样

[2, 2, 3, 4, 3, 4] --> [[2], [2], [3], [4], [3], [4], [2, 2], [2, 3], [3, 4], ...]

zltzlt

TJBEST 发表于 2020-1-22 14:19
[3,4,3,4]这样的数组 是不是存在两个[3,4]啊？还是算一个

两个 [3, 4]

阴阳神万物主

要去重另说，但应该不用去重。

1. def solve(lst):
   
2.     times = len(lst)
   
3.     res = lst[0]*times if times else 0
   
4.     try:
   
5.         lst.pop(0)
   
6.         res += lst[-1]*times
   
7.         lst.pop(-1)
   
8.     except:
   
9.         return res
   
10.     times += len(lst)
    
11.     for each in lst:
    
12.         res += each * times
    
13.     return res
    
14. if __name__ == '__main__':
    
15.     print('示例1 输出：',solve([1,2,3]))
    
16.     print('示例2 输出：',solve([1,2]))
    
17. 
    

复制代码

zltzlt

阴阳神万物主 发表于 2020-1-22 14:36
要去重另说，但应该不用去重。

解答错误

输入：[5, 2, 8, 4, 2]
输出：147
预期结果：155

Croper

1. def func312(l:list):
   
2.         left,right=1,len(l)
   
3.         ret=0
   
4.         for i in l:
   
5.                 ret+=i*left*right
   
6.                 left+=1
   
7.                 right-=1
   
8.         return ret

复制代码

Croper

写成一行的版本

1. def func312(l:list):
   
2.         return sum((i+1)*(len(l)-i)*l[i] for i in range(len(l)))

复制代码

Croper

看起来测试用例有点长。。。

kinkon

Croper 发表于 2020-1-22 14:51
写成一行的版本

只是佩服Croper老大的头脑

塔利班

1. def f312(x):
   
2.     s=sum(x)
   
3.     n=len(x)
   
4.     r=n*(n+1)//2*s
   
5.     coef=[(n-i)*(n-i-1)//2 for i in range(n)]
   
6.     for e in zip(x,coef,coef[::-1]):
   
7.         a,b,c=e
   
8.         r-=a*(b+c)
   
9.     return r

复制代码

kinkon

塔利班 发表于 2020-1-22 15:30

佩服...

塔利班

kinkon 发表于 2020-1-22 16:07
佩服...

没有，看了大神的感觉走了点弯路，，

TJBEST

我的代码总是很长，需要改进

1. def fun312(arr):
   
2.     def inner(n):
   
3.         down_gate = n - 1
   
4.         up_gate = N - n
   
5.         res = 0
   
6.         if down_gate <= up_gate:
   
7.             for i in range(0,down_gate):
   
8.                 res = res + (i + 1)*arr[i]
   
9.             for i in range(down_gate,up_gate + 1):
   
10.                 res = res + n * arr[i]
    
11.             for i in range(up_gate + 1,N):
    
12.                 res = res + (N - i) * arr[i]
    
13.             return res
    
14.         else:
    
15.             for i in range(0,up_gate):
    
16.                 res = res + (i + 1)*arr[i]
    
17.             for i in range(up_gate,down_gate + 1):
    
18.                 res = res + (N - n + 1) * arr[i]
    
19.             for i in range(down_gate + 1,N):
    
20.                 res = res + (N - i) * arr[i]
    
21.             return res
    
22.             
    
23.                
    
24.     if arr == []:
    
25.         return 0
    
26.    
    
27.     res = 0
    
28.     N = len(arr)
    
29.     for i in range(1,N + 1):
    
30.         res = res + inner(i)
    
31.     return res

复制代码

newbison

上一题不会，这一题我用递归来解决吧。实在看不懂你们的高深的方法。

1. def func312(array):
   
2.     l = len(array)
   
3.     res = 0
   
4.     if l == 1:
   
5.         res = array[0]
   
6.     elif l == 2:
   
7.         res = array[0] + array[1] + (array[0] + array[1])
   
8.     else:
   
9.         res = func312(array[:(l-1)])
   
10.         for i in range(0, l):
    
11.             res += add(array[i:])
    
12.     #递归法解决该问题，每添加一个元素，多出来的数组统统加一遍
    
13.     return res
    
14. 
    
15. def add(array):
    
16.     res = 0
    
17.     for each in array:
    
18.         res += each
    
19.     return res
    
20. 
    
21. 
    
22. print('array [1, 2, 3]的答案是: ', func312([1, 2, 3]))
    
23. print('array [1, 2]的答案是: ', func312([1, 2]))
    
24. print('array [5, 2, 8, 4, 2]的答案是: ', func312([5, 2, 8, 4, 2]))
    

复制代码

fan1993423

1. def fun312(lst):
   
2.     t=[]
   
3.     for i in range(1,len(lst)+1):
   
4.         for j in range(len(lst)-i+1):
   
5.             t.extend(lst[j:j+i])
   
6.     return sum(t)

复制代码

可能效率低哈