C++零基础算法100题 - 移除链表元素

题目链接

题目描述

解题思路

1. 如果说第一个元素就是要删除的元素。一直往后面遍历
2. 遍历这个head，直到cur==NULL
3. 用tmp变量临时保存cur->next的节点值
4. 如果说cur->next->val的值为传入进来的val。那么我们就移除这个val

代码实现

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode() : val(0), next(nullptr) {}
 *     ListNode(int x) : val(x), next(nullptr) {}
 *     ListNode(int x, ListNode *next) : val(x), next(next) {}
 * };
 */
class Solution {
public:
    ListNode* removeElements(ListNode* head, int val) {
        while(head!=NULL&&head->val==val){
            ListNode *tmp=head;
            head=head->next;
            delete tmp;
        }
        ListNode *cur=head;
        while(cur!=NULL&&cur->next!=NULL){
            // 如果说cur->next的val为要删除的值
            if(cur->next->val==val){
                ListNode *tmp=cur->next;
                cur->next=cur->next->next;
                delete tmp;
            }else{
                cur=cur->next;
            }
        }
        return head;
    }
};