题目链接:23.移除链表元素
题目:
给你一个链表的头节点 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
输出:[]
思路和算法:
两种方案:迭代和递归。
代码(c++):
方案一:迭代
//迭代
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode* sentinel = new ListNode(0, head); //定义一个哨兵
ListNode* node = sentinel;
while (node->next) {
if (node->next->val == val) {
node->next = node->next->next;
}
else {
node = node->next;
}
}
head = sentinel->next;
delete sentinel; //删除哨兵节点
return head;
}
};
方案二:递归
//递归
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
if (head == nullptr) return head;
head->next = removeElements(head->next, val);
return head->val == val ? head->next : head;
}
};
开心~今天头一回在vs上运行链表,之前都是直接在力扣上做的。虽然之前在力扣上还可以,但是有一个弊端,就是在面试的时候,如果面试官要求自己写一整套代码用于运行链表,自己却不会写创建链表和主函数这两块地方,这就很难受。