Python程式：按k大小分組反轉連結串列

假設我們有一個單鏈表，還有一個值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 的下一個節點 := node
  • prev := null，curr := null
  • lp := temp，lc := curr
  • cnt := k
  • 當curr不為空時，執行：
    • prev := null
    • 當cnt > 0且curr不為空時，執行：
      • following := curr 的下一個節點
      • curr 的下一個節點 := prev
      • prev := curr，curr := following
      • cnt := cnt - 1
    • lp 的下一個節點 := prev，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, ]

更新於：2020年10月20日

335 次瀏覽

啟動您的職業生涯

完成課程獲得認證

開始學習
廣告