算法打卡day13-删除链表的节点

描述

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

1.此题对比原题有改动

2.题目保证链表中节点的值互不相同

3.该题只会输出返回的链表和结果做对比，所以若使用 C 或 C++ 语言，你不需要 free 或 delete 被删除的节点

数据范围:

0<=链表节点值<=10000

0<=链表长度<=10000

示例1

输入：

{2,5,1,9},5

返回值：

{2,1,9}

说明：

给定你链表中值为 5 的第二个节点，那么在调用了你的函数之后，该链表应变为 2 -> 1 -> 9   

示例2

输入：

{2,5,1,9},1

返回值：

{2,5,9}

说明：

给定你链表中值为 1 的第三个节点，那么在调用了你的函数之后，该链表应变为 2 -> 5 -> 9   

代码实现

struct ListNode* deleteNode(struct ListNode* head, int val ) {
    if (head == NULL) {
        return NULL;
    }

    struct ListNode* pre = NULL;
    struct ListNode* node = head;
    struct ListNode* next = head->next;

    while (node) {
        int cur_val = node->val;
        if (cur_val == val) {
            if (pre != NULL) {
                pre->next = next;
            } else {
                head = next;
            }
            break;
        } else {
            pre = node;
            node = next;
            if (next) {
                next = next->next;
            }
        }
    }
    return head;
}