在链表中删除指定值的节点

在链表中删除指定值的节点

题目描述

给出一个链表和一个整数 num，输出删除链表中节点值等于 num 的节点之后的链表。

输入描述：

第一行一个整数 n，n 表示单链表的节点数量。

第二行 n 个整数表示单链表的各个节点的值。

第三行一个整数 num。

输出描述：

在给定的函数中返回指定链表的头指针。

示例1

输入

4 
1 2 3 4
3

输出

1 2 4

备注：

$1\le n\le 10^6$

$-10^6\le num,val\le 10^6$

---

题解：

正常的链表删除，不过有一个小技巧，创建一个哨兵节点，可以省很多事。

代码：

# include <bits/stdc++.h>
using namespace std;

struct list_node{
    int val;
    struct list_node * next;
};

list_node * input_list()
{
    int val, n;
    scanf("%d", &n);
    list_node * phead = new list_node();
    list_node * cur_pnode = phead;
    for (int i = 1; i <= n; ++i) {
        scanf("%d", &val);
        if (i == 1) {
            cur_pnode->val = val;
            cur_pnode->next = NULL;
        }
        else {
            list_node * new_pnode = new list_node();
            new_pnode->val = val;
            new_pnode->next = NULL;
            cur_pnode->next = new_pnode;
            cur_pnode = new_pnode;
        }
    }
    return phead;
}


list_node * remove_value(list_node * head, int num)
{
    //在下面完成代码
    list_node *dummy = new list_node();
    dummy->next = head;
    list_node *p = head, *pre = dummy;
    while (p) {
        if (p->val == num) {
            pre->next = p->next;
            delete p;
            p = pre->next;
        } else {
            pre = p;
            p = p->next;
        }
    }
    return dummy->next;
}

void print_list(list_node * head)
{
    while (head != NULL) {
        printf("%d ", head->val);
        head = head->next;
    }
    puts("");
}

int main ()
{
    list_node * head = input_list();
    int num;
    scanf("%d", &num);
    list_node * new_head = remove_value(head, num);
    print_list(new_head);
    return 0;
}