Leetcode面试题18 删除链表的节点 C++,Java,Python

Leetcode面试题18 删除链表的节点

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

博主Github：https://github.com/GDUT-Rp/LeetCode

题目：

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

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

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

示例 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 被删除的节点

解题思路：

方法一：快慢指针

直观想法

设置一个fast指针、slow指针都指向head，fast指针走 $k$ 步，然后两个指针都一步一步走，即可找到链表倒数第 $k$ 个节点。
使用双指针则可以不用统计链表长度。

C++

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* deleteNode(ListNode* head, int val) {
        if (head == nullptr) {
            return head;
        }
        ListNode *p = head, *prev = head;
        while (p->next != nullptr) {
            if (p->val == val) {
                p->val = p->next->val;
                p->next = p->next->next;
                return head;
            }
            prev = p;
            p = p->next;
        }
        // 待删除的节点在最后一个节点
        prev->next = nullptr;
        return head;
    }
};

Java

class Solution {
    public ListNode getKthFromEnd(ListNode head, int k) {
        ListNode former = head, latter = head;
        for(int i = 0; i < k; i++)
            former = former.next;
        while(former != null) {
            former = former.next;
            latter = latter.next;
        }
        return latter;
    }
}

Python

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def deleteNode(self, head: ListNode, val: int) -> ListNode:
        if not head:
            return head
        p, prev = head, head
        while p.next:
            if p.val == val:
                p.val = p.next.val
                p.next = p.next.next
                return head
            prev = p
            p = p.next
        # last one
        prev.next = p.next
        return head                

复杂度分析

时间复杂度：$O(n)$

空间复杂度：$O(1)$