203. 移除链表元素

题目

给你一个链表的头节点 head 和一个整数 val ，请你删除链表中所有满足 Node.val == val 的节点，并返回 新的头节点 。

示例 1：

输入：head = [1,2,6,3,4,5,6], val = 6
输出：[1,2,3,4,5]

示例 2：

输入：head = [], val = 1
输出：[]

示例 3：

输入：head = [7,7,7,7], val = 7
输出：[]

提示：

• 列表中的节点数目在范围 [0, 104] 内
• 1 <= Node.val <= 50
• 0 <= val <= 50

代码(解析在注释)

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     struct ListNode *next;
 * };
 */

typedef struct ListNode ListNode ;
struct ListNode* removeElements(struct ListNode* head, int val) {
    // 思路我们直接遍历链表，把不是val的数据放在一个新的链表中(c创建新的链表)
    // 创建一个新链表的头和尾
    ListNode* newhead, *newtail;
    // 初始化NULL
    newhead = newtail = NULL;
    // 创建一个小弟
    ListNode* tmp = head;
    // 利用循环遍历
    while (tmp) {
        // 判断节点是否为val,不是val就直接尾插到新的链表中
        if (tmp->val != val) {
            // 判断新链表是否为空，这里只需要判断head
            if (newhead == NULL) {
                // head 和
                // tail都指向第一个节点，所以这里就连续赋值tmp，就有了第一个节点
                newhead = newtail = tmp;
            }
            // 不是为空的情况就直接用tail->next
            else {
                newtail->next = tmp;
                newtail = newtail->next;
            }
        }
        tmp = tmp->next;
    }
    //链表有可能一开始就是NULL，所以这里需要判断一下链表是否为NULL
    if(newtail)//需要判断尾节点newtail
    newtail->next = NULL;
    return newhead;
}