24. 两两交换链表中的节点 中等
给定一个链表,两两交换其中相邻的节点,并返回交换后的链表。
你不能只是单纯的改变节点内部的值,而是需要实际的进行节点交换。
示例 1:
输入:head = [1,2,3,4]
输出:[2,1,4,3]
示例 2:
输入:head = []
输出:[]
示例 3:
输入:head = [1]
输出:[1]
提示:
- 链表中节点的数目在范围 [0, 100] 内
- 0 <= Node.val <= 100
进阶:你能在不修改链表节点值的情况下解决这个问题吗?(也就是说,仅修改节点本身。)
代码参考:
package main
import "fmt"
func main() {
fmt.Println(swapPairs(newList([]int{1})))
fmt.Println(swapPairs(newList([]int{1, 2})))
fmt.Println(swapPairs(newList([]int{1, 2, 3, 4})))
}
// 处理好 2 种异常 case
// 将两两节点视为内部短链表,交换两个节点后返回新的头节点(原第二节点),接到尾巴上即可
// 一般 case:pre->cur->next 经 swap() 调换后 next->pre,手动接上 pre->next->cur
func swapPairs(head *ListNode) *ListNode {
if head == nil || head.Next == nil { // 异常case: [] 或 [1]
return head
}
newHead := swap(head, head.Next)
pre := head
if pre.Next == nil || pre.Next.Next == nil { // 异常case: [1, 2]
return newHead
}
cur := pre.Next
next := cur.Next
for cur != nil && next != nil { // 一般case: 两两结对向后遍历
pre.Next = swap(cur, next) // 内部交换
if cur.Next == nil {
break
}
pre = cur
cur = cur.Next
next = cur.Next
}
return newHead
}
func swap(cur, next *ListNode) *ListNode {
cur.Next = next.Next
next.Next = cur
return next
}
最后编辑: kuteng 文档更新时间: 2021-06-05 10:16 作者:kuteng