[已解决]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

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

def func312(array):
    l = len(array)
    res = 0
    if l == 1:
        res = array[0]
    elif l == 2:
        res = array[0] + array[1] + (array[0] + array[1])
    else:
        res = func312(array[:(l-1)])
        for i in range(0, l):
            res += add(array[i:])
    #递归法解决该问题，每添加一个元素，多出来的数组统统加一遍
    return res

def add(array):
    res = 0
    for each in array:
        res += each
    return res 


print('array [1, 2, 3]的答案是: ', func312([1, 2, 3]))
print('array [1, 2]的答案是: ', func312([1, 2]))
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]

阴阳神万物主

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

def solve(lst):
    times = len(lst)
    res = lst[0]*times if times else 0
    try:
        lst.pop(0)
        res += lst[-1]*times
        lst.pop(-1)
    except:
        return res
    times += len(lst)
    for each in lst:
        res += each * times
    return res
if __name__ == '__main__':
    print('示例1 输出：',solve([1,2,3]))
    print('示例2 输出：',solve([1,2]))

zltzlt

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

解答错误

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

Croper

def func312(l:list):
        left,right=1,len(l)
        ret=0
        for i in l:
                ret+=i*left*right
                left+=1
                right-=1
        return ret

Croper

写成一行的版本

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

Croper

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

kinkon

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

只是佩服Croper老大的头脑

塔利班

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

kinkon

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

佩服...

塔利班

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

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

TJBEST

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

def fun312(arr):
    def inner(n):
        down_gate = n - 1
        up_gate = N - n
        res = 0
        if down_gate <= up_gate:
            for i in range(0,down_gate):
                res = res + (i + 1)*arr[i]
            for i in range(down_gate,up_gate + 1):
                res = res + n * arr[i]
            for i in range(up_gate + 1,N):
                res = res + (N - i) * arr[i]
            return res
        else:
            for i in range(0,up_gate):
                res = res + (i + 1)*arr[i]
            for i in range(up_gate,down_gate + 1):
                res = res + (N - n + 1) * arr[i]
            for i in range(down_gate + 1,N):
                res = res + (N - i) * arr[i]
            return res
            
                
    if arr == []:
        return 0
   
    res = 0
    N = len(arr)
    for i in range(1,N + 1):
        res = res + inner(i)
    return res

newbison

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

def func312(array):
    l = len(array)
    res = 0
    if l == 1:
        res = array[0]
    elif l == 2:
        res = array[0] + array[1] + (array[0] + array[1])
    else:
        res = func312(array[:(l-1)])
        for i in range(0, l):
            res += add(array[i:])
    #递归法解决该问题，每添加一个元素，多出来的数组统统加一遍
    return res

def add(array):
    res = 0
    for each in array:
        res += each
    return res 


print('array [1, 2, 3]的答案是: ', func312([1, 2, 3]))
print('array [1, 2]的答案是: ', func312([1, 2]))
print('array [5, 2, 8, 4, 2]的答案是: ', func312([5, 2, 8, 4, 2]))

fan1993423

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

可能效率低哈