刷题笔记-移除链表元素

题目描述

https://leetcode-cn.com/problems/remove-linked-list-elements/

解答

（哭了，大四时候看的数据结构还没忘没【捂脸】）

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        ListNode* list=head;//新链表的头
        ListNode* temp;//值为val的节点的上一个节点
        while(head!=NULL)
        {
        //如果当前节点的值为val，且当前节点是第一个节点，则更新链表头list
            if(head->val==val)
            {
                if(head == list)
                    list = head->next;
                else
                    temp->next = head->next;
            }
            //如果是中间节点或者链表最后一个节点，将上一个节点的next指针指向下一个节点
            else
            {
                temp = head;
            }
            head = head->next;
        }
        return list;
    }
};