1. 程式人生 > >LeetCode24 兩兩交換連結串列中的節點 Python3

LeetCode24 兩兩交換連結串列中的節點 Python3

給定一個連結串列,兩兩交換其中相鄰的節點,並返回交換後的連結串列。

示例:

給定 1->2->3->4, 你應該返回 2->1->4->3. 說明:

你的演算法只能使用常數的額外空間。 你不能只是單純的改變節點內部的值,而是需要實際的進行節點交換。

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def swapPairs(self, head):
        """
        :type head: ListNode
        :rtype: ListNode
        """
        new_head = ListNode(0)
        new_head.next = head
        pre = new_head
        while pre.next != None and pre.next.next != None:
            node1 = pre.next
            node2 = pre.next.next
            node3 = pre.next.next.next
            
            pre.next = node2
            node1.next = node3
            node2.next = node1
            
            pre = node1
        
        return new_head.next