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

Chương trình hợp nhất giữa các danh sách được liên kết bằng Python

Giả sử chúng ta có hai danh sách liên kết L1 và L2 có độ dài lần lượt là m và n, chúng ta cũng có hai vị trí a và b. Chúng ta phải xóa các nút khỏi L1 từ nút thứ a đến nút thứ b và hợp nhất L2 vào giữa.

Vì vậy, nếu đầu vào giống như L1 =[1,5,6,7,1,6,3,9,12] L2 =[5,7,1,6] a =3 b =6, thì đầu ra sẽ là [1, 5, 6, 5, 7, 1, 6, 9, 12]

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

  • head2:=L2, temp:=L2
  • trong khi tạm thời có nút tiếp theo, hãy thực hiện
    • temp:=tiếp theo của temp
  • tail2:=temp
  • số lượng:=0
  • tạm thời:=L1
  • end1:=null, start3:=null
  • trong khi tạm thời không null, hãy thực hiện
    • nếu số đếm giống như a-1, thì
      • end1:=temp
    • nếu số đếm giống như b + 1, thì
      • start3:=temp
      • ra khỏi vòng lặp
    • temp:=tiếp theo của temp
    • count:=count + 1
  • tiếp theo của end1:=head2
  • tiếp theo của tail2:=start3
  • trả về L1

Ví dụ

Hãy cùng chúng tôi xem cách triển khai sau để hiểu rõ hơn -

class ListNode:
   def __init__(self, data, next = None):
      self.val = data
      self.next = next

def make_list(elements):
   head = ListNode(elements[0])
   for element in elements[1:]:
      ptr = head
      while ptr.next:
         ptr = ptr.next
      ptr.next = ListNode(element)
   return head

def print_list(head):
   ptr = head
   print('[', end = "")
   while ptr:
      print(ptr.val, end = ", ")
      ptr = ptr.next
   print(']')

def solve(L1, L2, a, b):
   head2 = temp = L2
   while temp.next:
      temp = temp.next
   tail2 = temp

   count = 0
   temp = L1
   end1, start3 = None, None
   while temp:
      if count == a-1:
         end1 = temp
      if count == b+1:
         start3 = temp
         break
      temp = temp.next
      count += 1

   end1.next = head2
   tail2.next = start3
   return L1

L1 = [1,5,6,7,1,6,3,9,12]
L2 = [5,7,1,6]
a = 3
b = 6
print_list(solve(make_list(L1), make_list(L2), a, b))

Đầu vào

[1,5,6,7,1,6,3,9,12], [5,7,1,6], 3, 6

Đầu ra

[1, 5, 6, 5, 7, 1, 6, 9, 12, ]