Chương trình tìm số còn thiếu thứ k từ danh sách các phần tử trong Python

Giả sử chúng ta có một danh sách các số duy nhất được sắp xếp gọi là num và một số nguyên k, chúng ta phải tìm số còn thiếu thứ k từ phần tử đầu tiên của danh sách đã cho.

Vì vậy, nếu đầu vào là nums =[5,6,8,10,11], k =1, thì đầu ra sẽ là 9, vì 9 là số thứ hai (chỉ số 1) bị thiếu.

Để 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 của nums, hãy thực hiện

  • diff:=nums [i] - nums [i - 1] - 1

  • nếu k> =diff, thì

    • k:=k - khác biệt

  • nếu không,

    • trả về số [i - 1] + k + 1

• trả về số [-1] + k + 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)):
         diff = nums[i] - nums[i - 1] - 1
         if k >= diff:
            k -= diff
         else:
            return nums[i - 1] + k + 1
      return nums[-1] + k + 1

ob = Solution()
nums = [5,6,8,10,11]
k = 1
print(ob.solve(nums, k))

Đầu vào

[5,6,8,10,11], 1

Đầu ra

9