Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

โปรแกรมเพื่อย้อนกลับรายการที่เชื่อมโยงตามกลุ่มขนาด k ใน Python


สมมติว่าเรามีรายการที่เชื่อมโยงเพียงอย่างเดียว และอีกค่าหนึ่งคือ k เราต้องย้อนกลับทุก k กลุ่มโหนดที่อยู่ติดกัน

ดังนั้น หากอินพุตเป็นเหมือน List =[1,2,3,4,5,6,7,8,9,10], k =3 ผลลัพธ์จะเป็น [3, 2, 1, 6, 5 , 4, 9, 8, 7, 10, ]

เพื่อแก้ปัญหานี้ เราจะทำตามขั้นตอนเหล่านี้ -

  • tmp :=โหนดใหม่ที่มีค่า 0
  • ถัดไปของ tmp :=โหนด
  • ก่อนหน้า :=null, สกุลเงิน :=null
  • lp :=temp, lc :=curr
  • cnt :=k
  • ในขณะที่ curr ไม่เป็นค่าว่าง ให้ทำ
    • ก่อนหน้า :=null
    • ในขณะที่ cnt> 0 และสกุลเงินไม่เป็นค่าว่าง ให้ทำ
      • กำลังติดตาม :=ถัดไปของสกุลเงิน
      • ถัดไปของสกุลเงิน :=ก่อนหน้า
      • ก่อนหน้า :=สกุลเงิน, สกุลเงิน :=กำลังติดตาม
      • cnt :=cnt - 1
    • ถัดไปของ lp :=ก่อนหน้า ถัดไปของ lc :=curr
    • lp :=lc, lc :=curr
    • cnt :=k
  • ส่งคืนถัดไปของ tmp

ให้เราดูการใช้งานต่อไปนี้เพื่อความเข้าใจที่ดีขึ้น -

ตัวอย่าง

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(']')
class Solution:
   def solve(self, node, k):
      tmp = ListNode(0)
      tmp.next = node
      prev, curr = None, node
      lp, lc = tmp, curr
      cnt = k
      while curr:
         prev = None
         while cnt > 0 and curr:
            following = curr.next
            curr.next = prev
            prev, curr = curr, following
            cnt -= 1
         lp.next, lc.next = prev, curr
         lp, lc = lc, curr
         cnt = k
      return tmp.next
ob = Solution()
head = make_list([1,2,3,4,5,6,7,8,9,10])
print_list(ob.solve(head, 3))

อินพุต

[1,2,3,4,5,6,7,8,9,10], 3

ผลลัพธ์

[3, 2, 1, 6, 5, 4, 9, 8, 7, 10, ]