LeetCode热题Hot100 - 两两交换链表中的节点

咸鱼在鱼塘2024-04-16 20:24

一刷~

给你一个链表，两两交换其中相邻的节点，并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题（即，只能进行节点交换）。

思路：

使用两个指针p1/p2，分别记录需要交换的两个节点。另外使用一个指针last，记录两个节点的上一个节点。 last.next = p2，p1.next = p2.next（记录p2后面的值，如果直接p2.next = p1，则p2后面待交换的节点会丢失），last.next.next = p1，完成交换。last指针往后移两个，重新生成p1和p2。

python 复制代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, val=0, next=None):
#         self.val = val
#         self.next = next
class Solution:
    def swapPairs(self, head: Optional[ListNode]) -> Optional[ListNode]:
        if not head or not head.next:
            return head
        
        p = ListNode(0)
        p.next = head

        p1, p2, last = head, head.next, p

        while p1 and p2:
            last.next = p2
            p1.next = p2.next
            last.next.next = p1
            last = p1

            if last.next:
                p1 = last.next
            else:
                break
            if p1.next:
                p2 = p1.next
            else:
                break
        
        return p.next