删除链表的节点

510 字

3 分钟

删除链表的节点

2021-07-29

2026-10-12

数据结构与算法

「剑指Offer」Swift

本题较为简单，熟悉链表应该达到随手就能解出的程度

题目#

给定单向链表的头指针和一个要删除的节点的值，定义一个函数删除该节点。

返回删除后的链表的头节点。

注意：此题对比原题有改动

示例 1:

1
输入: head = [4,5,1,9], val = 5
2
输出: [4,1,9]
3
解释: 给定你链表中值为 5 的第二个节点，那么在调用了你的函数之后，该链表应变为 4 -> 1 -> 9.

示例 2:

1
输入: head = [4,5,1,9], val = 1
2
输出: [4,5,9]
3
解释: 给定你链表中值为 1 的第三个节点，那么在调用了你的函数之后，该链表应变为 4 -> 5 -> 9.

说明：

题目保证链表中节点的值互不相同若使用 C 或 C++ 语言，你不需要 free 或 delete 被删除的节点

来源：力扣（LeetCode）

链接：https://leetcode-cn.com/problems/shan-chu-lian-biao-de-jie-dian-lcof

著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

一、思路#

本题意在移除一个节点，即：

1 -> 2 -> 3 -> 4 -> 5

如果要移除3节点，就要将2节点的next指向4。所以在遍历链表的时候我们在找到移除目标的时候，需要直到上个节点是什么。

这里引入一个指针 lastNode 保存上个节点，遍历时 currentNode 作为指向当前节点的指针。

遇到目标节点后，将当前节点的 next 赋值给 lastNode 的 next，返回 head。

二、双指针解法#

LeetCode

1
func deleteNode(_ head: ListNode?, _ val: Int) -> ListNode? {
2
    var lastNode: ListNode?
3
    var currentNode = head
4

5
    while let node = currentNode {
6
        if node.val == val {
7
            if let last = lastNode {
8
                last.next = node.next
9
                return head
10
            }
11
            else {
12
                return node.next
13
            }
14
        }
15

16
        lastNode = node
17
        currentNode = node.next
18
    }
19

20
    return head
21
}

三、单指针解法#

1
func deleteNode2(_ head: ListNode?, _ val: Int) -> ListNode? {
2
    if head?.val == val {
3
        return head?.next
4
    }
5
    var currentNode = head
6
    while let node = currentNode?.next {
7
        if node.val == val {
8
            currentNode?.next = node.next
9
            return head
10
        }
11
        else {
12
            currentNode = currentNode?.next
13
        }
14
    }
15
    return head
16
}