题目描述
题目描述
给你一个链表的头节点 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
思路:
本题属于典型的链表删除题目,删除链表的核心代码为:
cur -> next = cur -> next -> next;
就是跳过下一个的节点,直接和下一个的下一个节点相连,但需要注意,C++需要手动释放内存。
为了方便处理第一个节点,可以设置一个虚拟头节点。
代码:
class Solution {
public:ListNode* removeElements(ListNode* head, int val) {ListNode* dummyHead = new ListNode(0); //虚拟头节点,用于处理第一个节点dummyHead -> next = head;ListNode* cur = dummyHead; //临时节点,用于遍历链表while (cur -> next != NULL) { //终止条件if (cur -> next -> val == val) { //删除节点ListNode* tmp = cur -> next; cur -> next = cur -> next -> next;delete tmp; //手动释放内存}else {cur = cur -> next; //遍历下一个节点}}ListNode* res = dummyHead -> next;delete dummyHead;return res;}
};
时间复杂度O(n):遍历一遍链表
空间复杂度O(1):只定义了常数个指针变量