LeetCode 代码随想录跟练 Day3
203.移除链表元素
题目描述:
给你一个链表的头节点 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
输出:[]
最直观的思路是类似于数组遍历,从头节点head处遍历到末尾(cur->next==nullptr),并在这个过程中删除对应值的节点。按照这个思路需注意:1、由于需获取当前节点的val和next属性,所以cur不能为nullptr;2、所需删除的值是通过当前节点获得的,由于是单向链表,需定义节点pre用于记录cur之前的节点。代码如下:
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode* cur = head;
ListNode* pre = nullptr;
while (cur != nullptr) {
ListNode* tmp = cur;
if (cur->val == val) {
if (pre == nullptr) {
cur = cur->next;
head = cur;
} else {
pre->next = cur->next;
cur = cur->next;
}
delete tmp;
} else {
pre = cur;
cur = cur->next;
}
}
return head;
}
};
可优化的地方:代码对当前节点是否是头节点做了额外处理,可改为使用一个dummy节点,从dummy节点出发,在条件判断中只需检查cur->next。代码如下:
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode dummy(0);
dummy.next = head;
ListNode* cur = &dummy;
while (cur->next) {
ListNode* tmp = cur->next;
if (tmp->val == val) {
cur->next = tmp->next;
delete tmp;
} else {
cur = tmp;
}
}
return dummy.next;
}
};
设计链表
题目描述:
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性:val 和 next 。val 是当前节点的值,next 是指向下一个节点的指针/引用。
如果是双向链表,则还需要属性 prev 以指示链表中的上一个节点。假设链表中的所有节点下标从 0 开始。
实现 MyLinkedList 类:
- MyLinkedList() 初始化 MyLinkedList 对象。
- int get(int index) 获取链表中下标为 index 的节点的值。如果下标无效,则返回 -1 。
- void addAtHead(int val) 将一个值为 val 的节点插入到链表中第一个元素之前。在插入完成后,新节点会成为链表的第一个节点。
- void addAtTail(int val) 将一个值为 val 的节点追加到链表中作为链表的最后一个元素。
- void addAtIndex(int index, int val) 将一个值为 val 的节点插入到链表中下标为 index 的节点之前。如果 index 等于链表的长度,那么该节点会被追加到链表的末尾。如果 index 比长度更大,该节点将 不会插入 到链表中。
- void deleteAtIndex(int index) 如果下标有效,则删除链表中下标为 index 的节点。
示例:
输入
[“MyLinkedList”, “addAtHead”, “addAtTail”, “addAtIndex”, “get”, “deleteAtIndex”, “get”]
[[], [1], [3], [1, 2], [1], [1], [1]]
输出
[null, null, null, null, 2, null, 3]
解释
MyLinkedList myLinkedList = new MyLinkedList();
myLinkedList.addAtHead(1);
myLinkedList.addAtTail(3);
myLinkedList.addAtIndex(1, 2); // 链表变为 1->2->3
myLinkedList.get(1); // 返回 2
myLinkedList.deleteAtIndex(1); // 现在,链表变为 1->3
myLinkedList.get(1); // 返回 3
实现思路和上一题类似,使用dummy作为虚拟节点,作为指向头节点的节点进行操作,代码如下:
class MyLinkedList {
public:
struct ListNode {
int val;
ListNode* next;
ListNode(int val) : val(val), next(nullptr) {}
};
MyLinkedList() {
_size = 0;
_dummyNode = new ListNode(0);
}
int get(int index) {
if (index >= _size) {
return -1;
}
ListNode* cur = _dummyNode->next;
while (index-- > 0) {
cur = cur->next;
}
return cur->val;
}
void addAtHead(int val) {
ListNode* newHead = new ListNode(val);
newHead->next = _dummyNode->next;
_dummyNode->next = newHead;
++_size;
}
void addAtTail(int val) {
ListNode* newTail = new ListNode(val);
ListNode* cur = _dummyNode;
while (cur->next) {
cur = cur->next;
}
cur->next = newTail;
++_size;
}
void addAtIndex(int index, int val) {
if (index > _size) return;
ListNode* newNode = new ListNode(val);
ListNode* cur = _dummyNode;
while (index-- > 0) {
cur = cur->next;
}
newNode->next = cur->next;
cur->next = newNode;
++_size;
}
void deleteAtIndex(int index) {
if (index >= _size) return;
ListNode* cur = _dummyNode;
while (index-- > 0) {
cur = cur->next;
}
ListNode* tmp = cur->next;
cur->next = tmp->next;
delete tmp;
--_size;
}
private:
int _size;
ListNode* _dummyNode;
};
其中addAtHead和addAtTail的部分可以使用addAtIndex进行实现来优化,如下:
...
void addAtHead(int val) {
addAtIndex(0, val);
}
void addAtTail(int val) {
addAtIndex(_size, val);
}
...
206.反转链表
题目描述:
给你单链表的头节点 head ,请你反转链表,并返回反转后的链表。
示例 1:
输入:head = [1,2,3,4,5]
输出:[5,4,3,2,1]
示例 2:
输入:head = [1,2]
输出:[2,1]
示例 3:
输入:head = []
输出:[]
直观的思路也是从头节点遍历到尾节点,每经过一个节点修改其next指向前一个节点。为实现这个指向修改需要:pre指针记录cur上一个节点以及临时指针tmp记录cur的下一个节点,即当前节点的前后两节点,然后修改并前进,代码如下:
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode* pre = nullptr;
ListNode* cur = head;
while (cur) {
ListNode* tmp = cur->next;
cur->next = pre;
pre = cur;
cur = tmp;
}
return pre;
}
};
另一种思路是递归的实现,将任务分解为:使当前节点指向前一个节点,接着返回以当前节点为头节点翻转转后的新链表头节点。也就是从头节点开始,将当前链表看作当前节点和当前节点的后续链表;将后续链表看作一个整体,定义函数为返回翻转后的链表的头节点,并使head指向pre,在当前的递归轮次可以视为这个后续链表已调用该函数完成翻转,且翻转后的尾节点已指向cur,使当前cur指向pre即可,如图
代码如下:
class Solution {
public:
ListNode* reverseList(ListNode* head) {
return reverse(nullptr, head);
}
ListNode* reverse(ListNode* pre, ListNode* head) {
if (!head) return pre;
ListNode* tmp = head->next;
head->next = pre;
return reverse(head, tmp);
}
};