Python程式:查詢兩個給定連結串列的並集
假設我們有兩個已排序的連結串列L1和L2,我們需要返回一個新的已排序連結串列,它是這兩個給定連結串列的並集。
因此,如果輸入類似於L1 = [10,20,30,40,50,60,70] L2 = [10,30,50,80,90],則輸出將為[10, 20, 30, 40, 50, 60, 70, 80, 90]
為了解決這個問題,我們將遵循以下步驟:
- 定義一個函式solve()。這將接收L1,L2作為引數。
- 如果L1為空,則
- 返回L2
- 如果L2為空,則
- 返回L1
- 如果L1的值小於L2的值,則
- res := L1
- res的下一個節點 := solve(L1的下一個節點, L2)
- 否則,如果L2的值小於L1的值,則
- res := L2
- res的下一個節點 := solve(L2的下一個節點, L1)
- 否則,
- res := L1
- res的下一個節點 := solve(L1的下一個節點, L2的下一個節點)
- 返回res
讓我們看看下面的實現來更好地理解:
示例
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, L1, L2): if not L1: return L2 if not L2: return L1 if L1.val < L2.val: res = L1 res.next = self.solve(L1.next, L2) elif L2.val < L1.val: res = L2 res.next = self.solve(L2.next, L1) else: res = L1 res.next = self.solve(L1.next, L2.next) return res ob = Solution() L1 = make_list([10,20,30,40,50,60,70]) L2 = make_list([10,30,50,80,90]) print_list(ob.solve(L1, L2))
輸入
[10,20,30,40,50,60,70], [10,30,50,80,90]
輸出
[10, 20, 30, 40, 50, 60, 70, 80, 90, ]
廣告