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

Chương trình tìm tổng của sự khác biệt tuyệt đối của mọi cặp trong danh sách được sắp xếp bằng Python

Giả sử chúng ta có một danh sách các số đã được sắp xếp gọi là num, chúng ta phải tìm tổng các hiệu số tuyệt đối giữa mọi cặp số trong danh sách đã cho. Ở đây chúng ta sẽ coi (i, j) và (j, i) là các cặp khác nhau. Nếu câu trả lời là rất lớn, hãy sửa đổi kết quả thành 10 ^ 9 + 7.

Vì vậy, nếu đầu vào là nums =[2, 4, 8], thì đầu ra sẽ là 24, là | 2 - 4 | + | 2 - 8 | + | 4 - 2 | + | 4 - 8 | + | 8 - 2 | + | 8 - 4 |.

Để 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

  • tổng:=0

  • đố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

    • tổng:=total + (i * nums [i] - (kích thước của nums - 1 - i) * nums [i]) mod m

  • return (tổng số 2 *) mod m

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):
      m = 10**9 + 7
      total = 0
      for i in range(len(nums)):
         total += (i*nums[i] - (len(nums) - 1 - i)*nums[i]) % m
      return (2*total) % m
ob = Solution()
nums = [2, 4, 8]
print(ob.solve(nums))

Đầu vào

[2, 4, 8]

Đầu ra

24