[力扣]面试题18. 删除链表的节点

删除给定值的节点

Posted by JinFei on April 5, 2020

题目描述

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

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

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

Example1:

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

Example2:

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

说明

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

解题思路

  • 删除给定的值
  • 注意,这个值可能重复多次
  • 需要判断下一个
/**
 * 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) {
        ListNode* preNode = new ListNode(-1);
        preNode -> next = head;
        ListNode* dummy = preNode;
        ListNode* cur = head;
        while(cur){
            ListNode* curNext = cur -> next;
            if(cur -> val == val && (curNext == nullptr || curNext -> val != val)){
                preNode -> next = cur -> next;
                preNode = cur -> next;
                cur = cur -> next;
            }
            else{
                preNode = cur;
                cur = curNext;
            }
        }
        return dummy -> next;
    }
};