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