【算法60天:Day3】203.移除链表元素、206.反转链表
题目
移除链表元素(203)
删除链表中等于给定值 val 的所有节点。
链表https://leetcode.cn/problems/remove-linked-list-elements/
解题思路
虚拟头节点
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
// 删除头结点
while (head != NULL && head->val == val) { // 注意这里不是if
ListNode* tmp = head;
head = head->next;
delete tmp;
}
// 删除非头结点
ListNode* cur = head;
while (cur != NULL && cur->next!= NULL) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
return head;
}
};
反转链表(206)
给你单链表的头节点 head ,请你反转链表,并返回反转后的链表。
链接https://leetcode.cn/problems/reverse-linked-list/
解题思路
自己的思路是再定义一个新的链表,实现链表元素的反转,但其实这是对内存空间的浪费。
双指针法
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode* temp; // 保存cur的下一个节点
ListNode* cur = head;
ListNode* pre = NULL;
while(cur) {
temp = cur->next; // 保存一下 cur的下一个节点,因为接下来要改变cur->next
cur->next = pre; // 翻转操作
// 更新pre 和 cur指针
pre = cur;
cur = temp;
}
return pre;
}
};