Python反轉連結串列內部節點的程式
假設我們有一個連結串列,我們還有兩個值i和j,我們必須反轉從第i個到第j個節點的連結串列。最後返回更新後的列表。
因此,如果輸入像[1,2,3,4,5,6,7,8,9] i = 2 j = 6,則輸出將為[1, 2, 7, 6, 5, 4, 3, 8, 9, ]
為了解決這個問題,我們將遵循以下步驟
- prev_head := 建立一個值為null並指向節點的連結串列節點
- prev := prev_head, curr := node
- 迭代所有從0到i的值,執行:
- prev := curr, curr := curr的下一個節點
- rev_before := prev, rev_end := curr
- 迭代所有從0到(j - i)的值,執行:
- tmp := curr的下一個節點
- curr的下一個節點 := prev
- prev, curr := curr, tmp
- rev_before的下一個節點 := prev, rev_end.next := curr
- 返回prev_head的下一個節點
讓我們看看下面的實現來更好地理解
示例
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, i, j): prev_head = ListNode(None, node) prev, curr = prev_head, node for _ in range(i): prev, curr = curr, curr.next rev_before, rev_end = prev, curr for _ in range(j - i + 1): tmp = curr.next curr.next = prev prev, curr = curr, tmp rev_before.next, rev_end.next = prev, curr return prev_head.next ob = Solution() head = make_list([1,2,3,4,5,6,7,8,9]) i = 2 j = 6 print_list(ob.solve(head, i, j))
輸入
[1,2,3,4,5,6,7,8,9], 2, 6
輸出
[1, 2, 7, 6, 5, 4, 3, 8, 9, ]
廣告