给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
示例:
输入:[1,2,3,4]
输出:[2,1,4,3]
题干比较简单,理解起来没有什么难度,就是将一个链表中的节点两两交换,具体思路如《代码随想录》所示,因此主要对每行代码进行解释。
完整代码如下:
# 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: ListNode) -> ListNode:
dummy_head = ListNode(next=head)
current = dummy_head
# 必须有cur的下一个和下下个才能交换,否则说明已经交换结束了
while current.next and current.next.next:
temp = current.next # 防止节点修改
temp1 = current.next.next.next
current.next = current.next.next
current.next.next = temp
temp.next = temp1
current = current.next.next
return dummy_head.next
首先,定义一个虚拟头结点,将其指向真实头结点head,接着将dummy_head赋值给current。
设计循环,保证当前指针的下一个节点与下下个节点存在时才存在循环。
temp = current.next # 防止节点修改
temp1 = current.next.next.next
与之前学习中的一样,为防止后面代码编写时对后续节点进行修改,因此先保存当前指针节点的下个节点temp = current.next和下下个节点temp1 = current.next.next.next
在第一次循环中,current代表当前指针在虚拟头节点处,因此代码current.next = current.next.next指的是将current指向current.next.next(即2),如下图所示。
代码到这里的时候,current依旧是虚拟头结点,但current.next已经发生的改变,现在的current.next是2,接着下面的代码current.next.next = temp就是将current.next(即2)的下一个节点指向temp(即循环一开始保存的两两交换节点的前一个节点,本次循环中指的是1),如下图所示。
接下来的代码
temp.next = temp1
current = current.next.next
意思是将temp(即1)指向下一个节点temp1(循环一开始保存的 current.next.next.next,即3),将current.next.next(即1)赋值给current,完成第一次循环,接着按以上步骤完成所有循环。
完成所有循环后,return至dummy_head的下一个节点,即以真实头结点为起点的链表。