一、题目
两两交换链表中的节点
给你一个链表,两两交换其中相邻的节点,并返回交换后链表的头节点。你必须在不修改节点内部的值的情况下完成本题(即,只能进行节点交换)。
示例 1:
输入:head = [1,2,3,4]
输出:[2,1,4,3]
示例 2:
输入:head = []
输出:[]
示例 3:
输入:head = [1]
输出:[1]
提示:
- 链表中节点的数目在范围 [0, 100] 内
- 0 <= Node.val <= 100
二、解题思路
题目提到需要将相邻的两个节点进行位置交换,所以我们需要记录被交换的两个节点,以及被交换节点的前序节点和后续节点。
定义如下四个节点:
pre-指向当前节点的前序节点
cur - 指向当前节点(需要被交换的第一个节点)
next - 指向当前节点下一个节点(需要被交换的第二个节点)
suf - 需要被交换两个节点的后续节点
将两个节点进行交换,需要将两个节点之间的联系以及前后关联节点的联系都进行更换,即:
pre.Next = next
next.Next = cur
cur.Next = suf
交换完毕后,将pre节点移动到cur节点位置即可开始下一轮移动,循环的终止条件是需要交换的两个节点任意一个节点为nil。
核心交换逻辑处理完后,还有一个问题需要处理:函数需要返回的是交换完毕后的头节点,这个头节点需要怎么获取呢?
在开始交换节点之前,用一个变量保存待交换的链表头节点即可
可以定义一个临时节点,next指针指向当前链表head,当当前链表交换完毕后,直接返回该临时节点的next指针即可。这个也是链表题中最常见的套路之一
headNode := &ListNode{
Val: 0,
Next: head,
}
xxxx swap node
return headNode.Next
具体的流程示意图如下:
三、代码实现
使用golang实现代码如下:
type ListNode struct {
Val int
Next *ListNode
}
func swapPairs(head *ListNode) *ListNode {
headNode := &ListNode{
Val: 0,
Next: head,
}
pre := headNode
for pre.Next != nil && pre.Next.Next != nil {
cur := pre.Next
next := cur.Next
suf := next.Next
pre.Next = next
next.Next = cur
cur.Next = suf
pre = cur
}
return headNode.Next
}
提交结果: