Computer >> Máy Tính >  >> Lập trình >> Python

Chương trình tìm tổng của tất cả các danh sách con liền kề trong Python

Giả sử chúng ta có một danh sách các số được gọi là num, bây giờ hãy xem xét mọi mảng con liền kề. Tính tổng từng mảng con này và trả về tổng của tất cả các giá trị này. Cuối cùng, sửa đổi kết quả theo 10 ** 9 + 7.

Vì vậy, nếu đầu vào là nums =[3, 4, 6], thì đầu ra sẽ là 43, vì Chúng ta có các mảng con sau - [3] [4] [6] [3, 4] [4, 6] [3, 4, 6] Tổng của tất cả những thứ này là 43.

Để giải quyết vấn đề này, chúng tôi sẽ làm theo các bước sau -

  • N:=kích thước của nums
  • ans:=0
  • đối với tôi trong phạm vi từ 0 đến kích thước là num, hãy thực hiện
    • n:=nums [i]
    • ans:=ans + (i + 1) * (N-i) * n
  • return (ans mod 1000000007)

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

Ví dụ

class Solution:
   def solve(self, nums):
      N=len(nums)
      ans=0
      for i in range(len(nums)):
         n=nums[i]
         ans += (i+1) * (N-i) * n
      return ans%1000000007
ob = Solution()
print(ob.solve([3, 4, 6]))

Đầu vào

[3, 4, 6]

Đầu ra

43