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

Chương trình tìm các khoảng liền kề của một mảng duy nhất trong Python

Giả sử chúng ta có một danh sách các số duy nhất được gọi là nums. Chúng ta phải tìm một ma trận 2D được sắp xếp gồm các số trong đó mỗi danh sách đại diện cho một số tổng hợp khoảng bao gồm các số liền kề nhau.

Vì vậy, nếu đầu vào giống như nums =[10, 11, 12, 15, 16, 17, 28, 30], thì đầu ra sẽ là [[10, 12], [15, 17], [28, 28] , [30, 30]], như trong danh sách [10 đến 12], [15 đến 17] liền kề nhau, 28 và 30 ở đó, chúng được biểu thị là [28 đến 28] và [30 đến 30].

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

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

  • chèn vô cùng vào cuối nums

  • ans:=một danh sách mới


  • l:=nums [0]

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

    • nếu nums [i] không giống nums [i-1] + 1, thì

      • chèn [l, nums [i-1]] vào cuối ans

      • l:=nums [i]

  • trả lại ans

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):
   nums.sort()
   nums.append(1e9)
   ans=[]
   l=nums[0]
   for i in range(1,len(nums)):
      if nums[i] != nums[i-1] + 1:
         ans.append([l, nums[i-1]])
         l=nums[i]
   return ans
ob = Solution()
nums = [10, 11, 12, 15, 16, 17, 28, 30]
print(ob.solve(nums))

Đầu vào

[10, 11, 12, 15, 16, 17, 28, 30]

Đầu ra

[[10, 12], [15, 17], [28, 28], [30, 30]]