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, ]
廣告
資料結構
網路
RDBMS
作業系統
Java
iOS
HTML
CSS
Android
Python
C語言程式設計
C++
C#
MongoDB
MySQL
Javascript
PHP