Chương trình tìm tổng chiều rộng của tất cả các dãy con của danh sách số trong Python

Giả sử chúng ta có một danh sách các số được gọi là nums, Chiều rộng của một dãy số là hiệu giữa số lớn nhất và số nhỏ nhất trong dãy. Chúng ta phải tìm tổng chiều rộng của tất cả các dãy con của số. Nếu câu trả lời là rất lớn thì hãy sửa đổi kết quả bằng 10 ^ 9 + 7.

Vì vậy, nếu đầu vào là nums =[7, 4, 9], thì đầu ra sẽ là 15, vì chúng ta có các dãy con như:[7], [4], [9], [7, 4], [ 7, 9], [4, 9], [7, 4, 9] và Vì vậy, chiều rộng là 0, 0, 0, 3, 2, 5, 5, do đó, lấy 15.

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

• m:=10 ^ 9 + 7

• sắp xếp số lượng danh sách

• ans:=0

• power:=một danh sách có kích thước giống như (nums + 1) và điền bằng 1

• đối với tôi trong phạm vi từ 1 đến kích thước của nums + 1, hãy thực hiện

  • power [i]:=power [i - 1] * 2 mod m

  • đối với tôi trong phạm vi từ 0 đến kích thước của nums, hãy thực hiện

    • dương:=(power [i] - 1) * nums [i]

    • âm:=(power [kích thước của nums - i - 1] - 1) * nums [i]

    • ans:=(ans + dương - âm) mod m

• trả lại ans

Ví dụ

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

class Solution:
   def solve(self, nums):
      m = 10**9 + 7
      nums.sort()
      ans = 0
      power = [1] * (len(nums) + 1)
      for i in range(1, len(nums) + 1):
         power[i] = power[i - 1] * 2 % m
      for i in range(0, len(nums)):
         positive = (power[i] - 1) * nums[i]
         negative = (power[len(nums) - i - 1] - 1) * nums[i]
         ans = (ans + positive - negative) % m
      return ans
ob = Solution()
nums = [7, 4, 9]
print(ob.solve(nums))

Đầu vào

[7, 4, 9]

Đầu ra

15