Tiền tố K trong Python

Giả sử chúng ta có một danh sách các số num và một số nguyên k, chúng ta phải tìm giá trị cực đại thứ i trong đó nums [0] + nums [1] + ... + nums [i] ≤ k. Chúng tôi sẽ trả về -1 nếu không có tôi hợp lệ.

Vì vậy, nếu đầu vào là nums =[4, -7, 5, 2, 6], k =5, thì đầu ra sẽ là 3, chỉ số là 3 như thể chúng ta thêm 4 + (- 7) + 5 + 2 =4, con số này nhỏ hơn k, nếu chúng ta thêm phần tử cuối cùng thì nó sẽ không còn nhỏ hơn k, vì vậy chỉ số là 3.

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

• đối với tôi trong phạm vi từ 1 đến kích thước là nums - 1, thực hiện
  • nums [i]:=nums [i] + nums [i-1]
• đối với tôi trong phạm vi kích thước của nums -1 đến -1, giảm đi 1, thực hiện
  • nếu nums [i] <=k, thì
    • trả lại tôi
• trả về -1

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, k):
      for i in range(1,len(nums)):
         nums[i]+=nums[i-1]
         for i in range(len(nums)-1,-1,-1):
            if nums[i]<=k:
                return i
      return -1
ob = Solution()
nums = [4, -7, 5, 2, 6]
k = 5
print(ob.solve(nums, k))

Đầu vào

[4, -7, 5, 2, 6], 5

Đầu ra

3