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

Chương trình đóng gói các phần tử liên tiếp giống nhau vào danh sách con bằng Python

Giả sử chúng ta có một danh sách các số num, chúng ta sẽ đóng gói các phần tử liên tiếp có cùng giá trị vào danh sách con. Chúng tôi phải lưu ý rằng chỉ có một lần xuất hiện trong danh sách, nó vẫn phải nằm trong danh sách con của chính nó.

Vì vậy, nếu đầu vào giống như nums =[5, 5, 2, 7, 7, 7, 2, 2, 2, 2], thì đầu ra sẽ là [[5, 5], [2], [7, 7, 7], [2, 2, 2, 2]]

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

  • nếu nums trống, thì
    • trả lại một danh sách mới
  • kết quả:=một danh sách với một danh sách khác chứa nums [0]
  • j:=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], thì
      • chèn một danh sách mới vào cuối kết quả
      • j:=j + 1
    • chèn nums [i] vào cuối kết quả [j]
  • trả về kết quả

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):
      if not nums:
         return []
      result = [[nums[0]]]
      j = 0
      for i in range(1, len(nums)):
         if nums[i] != nums[i - 1]:
            result.append([])
            j += 1
            result[j].append(nums[i])
      return result
ob = Solution()
nums = [5, 5, 2, 7, 7, 7, 2, 2, 2, 2]
print(ob.solve(nums))

Đầu vào

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

Đầu ra

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