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

Chương trình tìm tổng tối đa bằng cách thực hiện nhiều nhất k phép toán phủ định trong Python

Giả sử chúng ta có một danh sách các phần tử được gọi là nums, chúng ta cũng có một giá trị khác là k. Bây giờ chúng ta hãy xem xét một phép toán trong đó chúng ta chọn một phần tử từ nums và phủ định nó. Chúng ta có thể thực hiện đúng k số phép toán. Chúng tôi phải tìm tổng kết quả tối đa có thể được tạo ra.

Vì vậy, nếu đầu vào giống như nums =[2, 1, -6, -2] k =3, thì đầu ra sẽ là 9, nếu chúng ta phủ định -6 và -2 và 1 sẽ nhận được [2, -1, 6 , 2] và tổng của nó là 9.

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

  • nếu n giống 0 thì

    • trả về 0

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

  • đối với idx trong phạm vi 0 đến n - 1, thực hiện

    • nếu nums [idx] <0 và k> 0, thì

      • k:=k - 1

      • nums [idx]:=-nums [idx]

  • nếu k là số lẻ thì

    • return (tổng tất cả các phần tử có trong nums) - (tối thiểu 2 * nums)

  • trả về tổng của tất cả các phần tử có trong nums

Ví dụ

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

def solve(nums, k):
   n = len(nums)
   if n == 0:
      return 0

   nums.sort()
   for idx in range(n):
      if nums[idx] < 0 and k > 0:
         k -= 1
         nums[idx] *= -1

   if k & 1 == 1:
      return sum(nums) - 2 * min(nums)

   return sum(nums)

nums = [2, 1, -6, -2]
k = 3
print(solve(nums, k))

Đầu vào

[2, 1, -6, -2], 3

Đầu ra

9