Go&Java算法之删除链表的节点

580 阅读2分钟

这是我参与11月更文挑战的第10天,活动详情查看:2021最后一次更文挑战

删除链表的节点

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

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

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

示例 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.  

说明:

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

题解

方法一:双指针——Java

思路很简单就是找到需要删除节点的前一个节点,使用first.next = first.next.next删除即可。 有几处细节需要注意一下:

1.使用虚拟头节点dummyNode,也就是在头节点前再加一个节点,这样当需要删除头节点时,操作和后面的节点一样

2.使用指针first遍历的时候,循环条件是first != null && first.next != null,如果不加first != null, 当需要删除的是最后一个节点时,就会有空指针异常

class Solution {
    public ListNode deleteNode(ListNode head, int val) {
        if(head == null) {
            return head;
        }
        ListNode cur = head;
        ListNode pre = new ListNode(0);
        pre.next = cur;
        ListNode result = pre;
        while(cur != null) {
            if(cur.val == val) {
                pre.next = cur.next;
                break;
            }
            pre = cur;
            cur = cur.next;
        }
        return result.next;
    }
}

时间复杂度:O(N)

空间复杂度:O(1)

方法一:双指针——Go

设置两个指针分别指向头节点,pre (要删除节点的前一节点)和 cur (当前节点);

遍历整个链表,查找节点值为 val 的节点,找到了就删除该节点,否则继续查找。

2.1. 找到,将当前节点的前驱节点(pre 节点或者说是之前最近一个值不等于 val 的节点)连接到当前节点(cur 节点)的下一个节点(pre->next = cur->next)。

2.2. 没找到,继续遍历(cur = cur->next),更新最近一个值不等于 val 的节点(pre = cur)。

func deleteNode(head *ListNode, val int) *ListNode {
    if head == nil {
        return nil
    }

    if head.Val == val {
        return head.Next
    }

    pre, cur := head, head
    for cur != nil && cur.Val != val {
        pre, cur = cur, cur.Next
    }

    if cur != nil {
        pre.Next = cur.Next
    }

    return head
}