[LeetCode](面试题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.

说明:

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

解题思路

算法步骤:
特例处理: 当应删除头节点 head 时,直接返回 head.next 即可。
初始化: cur = head。
定位节点: 当 cur.next 非空,并且 cur.next 的节点值不等于 val 时进入循环,跳到下一节点:cur = cur.next 。
当 cur.next 非空(即不是尾节点末尾时),表示找到了要删除的节点,那么就更改引用关系:执行 cur.next = cur.next.next。(若跳出循环时 cur.next = null,代表链表中不包含值为 val 的节点。)
返回值: 返回链表头部节点 head 即可。

复杂度分析:
时间复杂度:O(N)。
空间复杂度:O(1)。

注:也可以在链表前面加一个伪头指针,这样就不用再判断原来的第一个头节点是否为要删除的节点了。

代码

解法一:不添加伪头指针

/**
 * 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.val == val){
            return head.next;
        }
        ListNode cur = head;
        while(cur.next != null && cur.next.val != val){
            cur = cur.next;
        }
        if(cur.next != null){
            cur.next = cur.next.next;
        }
        return head;
    }
}

解法二:添加伪头指针

/**
 * 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) {
        ListNode dummy = new ListNode(0);
        dummy.next = head;
        head = dummy;
        while(head.next != null && head.next.val != val){
            head = head.next;
        }
        if(head.next != null){
            head.next = head.next.next;
        }
        return dummy.next;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章