剑指Offer.18 删除链表的节点

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

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

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

示例 1:

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

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

说明：题目保证链表中节点的值互不相同

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shan-chu-lian-biao-de-jie-dian-lcof

解题思路

分情况讨论

如果要删除的节点刚好是头结点
1. 如果只有一个节点, 直接返回空链表
2. 如果还有后继, 则直接返回后继作为新头节点
如果要删除的节点不是头结点
1. 遍历链表, 直到找到要删除的节点的前一个节点
2. 跳出循环时有两种情况, 找到了要删除的节点以及找不到
  1. 找到了则当前节点的下一个节点不为空, 此时再判断要删除的节点是否为尾节点
    1. 如果是尾节点则让当前节点的next置为空
    2. 如果不是尾节点则cur.next = cur.next.next
  2. 找不到要删除的节点则当前节点已经是尾节点, 则返回null

代码

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
class Solution {
    public ListNode deleteNode(ListNode head, int val) {
        if (head == null) {
            return null;
        }
        // 如果要删除的节点刚好是头结点
        if (head.val == val) {
            // 并且只有一个节点, 直接返回空链表
            if (head.next == null) {
                return null;
            }
            // 如果还有后继, 则直接返回后继作为新头节点
            else {
                return head.next;
            }
        }
        ListNode cur = head;
        // 循环遍历直到找到要删除的节点的前一个节点
        while (cur.next != null && cur.next.val != val) {
            cur = cur.next;
        }
        // 跳出循环后如果当前节点的next为空, 说明链表中没有要删除的节点
        if (cur.next == null) {
            return null;
        }
        // 不为空则说明找到了节点, 所以提前跳出了循环
        else {
            // 如果要删除的节点不为尾节点, 则将当前节点的next赋值为要删除的节点的next
            if (cur.next.next != null) {
                cur.next = cur.next.next;
            }
            // 否则则是要删除尾节点, 则当前节点的next置空
            else {
                cur.next = null;
            }
        }
        return head;
    }
}