【leetcode】移除链表元素c++

题目描述:

给你一个链表的头节点 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 <= k <= 50

代码:

/**
 * 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) {
    
    
        if(head==NULL)return NULL;
        ListNode* node =  new ListNode(0);
        node->next=head; //首先把头节点加进去
        ListNode* temp=node;
        while(temp->next!=NULL){
    
    
            if(temp->next->val==val){
    
    
                temp->next=temp->next->next;
            }
            else if(temp->next->val!=val){
    
    
                temp=temp->next;
            }
        }
        return node->next;
    }
};

开始时,用new ListNode()设置虚拟虚拟节点作为真正头节点前的节点node。 node->next指向这正头节点,最终返回node->next。

猜你喜欢

转载自blog.csdn.net/qq_40315080/article/details/116059676
今日推荐