Python程式:查詢給定連結串列的摺疊連結串列


假設我們有一個連結串列。我們需要取連結串列的前半部分,並將後半部分摺疊過來,然後將交叉節點合併並求和。最後,我們需要返回結果連結串列的頭節點。

因此,如果輸入類似於 [5, 8, 1, 2, 4, 7, 5],則輸出將為 [2, 5, 15, 10]

為了解決這個問題,我們將遵循以下步驟:

  • temp := 0
  • ptr := 節點
  • 當 ptr 不為空時,執行以下操作:
    • temp := temp + 1
    • ptr := ptr 的下一個節點
  • t := temp / 2 的商
  • m := 節點
  • stk := 新建一個棧
  • 當 t 不為零時,執行以下操作:
    • 將 m 的值壓入 stk
    • tmp := m 的下一個節點
    • m 的下一個節點 := null
    • m := tmp
    • t := t - 1
  • 節點 := m
  • 如果 temp 為偶數,則:
    • m := m 的下一個節點
  • 當 m 不為空時,執行以下操作:
    • m 的值 := m 的值 + 棧頂元素。
    • 從 stk 中彈出元素
    • m := m 的下一個節點
  • 返回節點

讓我們看看下面的實現,以便更好地理解:

示例

 即時演示

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):
      temp = 0
      ptr = node
      while ptr:
         temp += 1
         ptr = ptr.next
      t = temp // 2
      m = node
      stk = []
      while t:
         stk.append(m.val)
         tmp = m.next
         m.next = None
         m = tmp
         t -= 1
      node = m
      if temp % 2 != 0:
         m = m.next
      while m:
         m.val += stk.pop()
         m = m.next
      return node
ob = Solution()
head = make_list([5,8,1,2,4,7,5])
print_list(ob.solve(head))

輸入

[5,8,1,2,4,7,5]

輸出

[2, 5, 15, 10, ]

更新時間: 2020年11月19日

224 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.