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

Chương trình để tối đa hóa giá trị nhỏ nhất sau khi tăng K danh sách con bằng Python

Giả sử chúng ta có một danh sách các số được gọi là num và hai giá trị, kích thước và k. Bây giờ, giả sử có một phép toán trong đó chúng ta lấy một danh sách con liền kề có kích thước chiều dài và tăng từng phần tử lên từng phần tử một. Chúng ta có thể thực hiện thao tác này k lần, chúng ta phải tìm giá trị nhỏ nhất lớn nhất có thể trong nums.

Vì vậy, nếu đầu vào là nums =[2, 5, 2, 2, 7], size =3, k =2, thì đầu ra sẽ là 3, vì chúng ta có thể tăng [2, 5, 2] để có được [ 3, 6, 3, 2, 7] rồi tăng [6, 3, 2] để có [3, 7, 4, 3, 7], tối thiểu là 3

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

  • Xác định một hàm khả thi (). Điều này sẽ đạt được mục tiêu
  • sự kiện:=Một danh sách kích thước N và điền bằng 0
  • di chuyển:=0, s:=0
  • đối với tôi trong phạm vi từ 0 đến N, thực hiện
    • s:=s + sự kiện [i]
    • delta:=target - (A [i] + s)
    • nếu delta> 0, thì
      • di chuyển:=di chuyển + delta
      • s:=s + delta
      • nếu i + size
      • sự kiện [i + size]:=sự kiện [i + size] - delta
  • trả về true khi di chuyển <=K
  • Từ phương pháp chính, hãy thực hiện như sau
  • N:=kích thước của A
  • left:=0, right:=10 ^ 10
  • khi left
  • giữa:=(trái + phải + 1) / 2
  • nếu có thể (giữa), thì
    • left:=mid
  • nếu không,
    • right:=mid - 1
  • quay lại bên trái
  • 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, A, size, K):
          N = len(A)
       def possible(target):
          events = [0] * N
          moves = s = 0
          for i in range(N):
             s += events[i]
             delta = target - (A[i] + s)
             if delta > 0:
                moves += delta
                s += delta
                if i + size < N:
                   events[i + size] -= delta
                   return moves <= K
                   left, right = 0, 10 ** 10
                   while left < right:
                      mid = (left + right + 1)//2
                   if possible(mid):
                      left = mid
                   else:
                      right = mid - 1
          return left
    ob = Solution()
    nums = [2, 5, 2, 2, 7]
    size = 3
    k = 2
    print(ob.solve(nums, size, k))

    Đầu vào

    [2, 5, 2, 2, 7], 3, 2

    Đầu ra

    3