代码随想录第三天 | 链表理论基础 | 203.移除链表元素 | 707.设计链表 | 206.反转链表
今天的主要内容是链表的相关操作。链表是一种十分基础且重要的数据结构,主要依靠指针进行操作,所以更加注重内存管理的问题。做题的时候就能感觉到,自己想当然设计的结构很容易就把边界条件给忽略了。
203 :移除链表元素
题目链接:203. 移除链表元素 - 力扣(LeetCode)
最重要的一点就是当头结点满足删除条件时如何处理。LeetCode给出的测试样例{7, 7, 7, 7},val = 7是个很好的例子,既要考虑删除头结点,也要考虑返回空指针。因为要删除节点,所以如果头结点就满足条件,要么特殊处理,然后就是设置一个虚拟头结点,来处理后续节点。从后面的题能发现,虚拟头结点是链表操作比较常用的一个方法。
链表使用LeetCode默认定义的结构。
struct ListNode{
int val;
ListNode * next;
ListNode(): val(0), next(nullptr){};
ListNode(int n) : val(n){};
ListNode(int n, ListNode * next): val(n), next(next){};
};
使用虚拟头结点的方法,使用c++的话要主要delete符合条件的ListNode,虽然LeetCode对此并没有严格要求,但是要自己要注意养成好习惯。
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode * dummyHead = new ListNode(0);
dummyHead->next = head;
ListNode * cur = dummyHead;
while(cur -> next != nullptr) {
if(cur -> next -> val == val) {
ListNode * temp = cur -> next;
cur -> next = cur -> next -> next;
delete temp;
} else {
cur = cur -> next;
}
}
head = dummyHead->next;
delete dummyHead;
return head;
}
};
707 设计链表
题目链接:707. 设计链表 - 力扣(LeetCode)
感觉这道题对于算法的考察就比较弱了,主要是考虑容错和设计合理的数据结构。比如要考虑index是否是一个合理的输入,如果不合理,如何处理的问题。
很好的链表基础题。其中使用了虚拟头结点来简化操作,基本覆盖了链表的增删改查操作。对着卡哥的视频学了一遍,感觉还是没有完全掌握,自己要多看看。
class MyLinkedList {
public:
// 定义链表节点结构体
struct LinkedNode {
int val;
LinkedNode* next;
LinkedNode(int val):val(val), next(nullptr){}
};
// 初始化链表
MyLinkedList() {
_dummyHead = new LinkedNode(0); // 这里定义的头结点 是一个虚拟头结点,而不是真正的链表头结点
_size = 0;
}
// 获取到第index个节点数值,如果index是非法数值直接返回-1, 注意index是从0开始的,第0个节点就是头结点
int get(int index) {
if (index > (_size - 1) || index < 0) {
return -1;
}
LinkedNode* cur = _dummyHead->next;
while(index--){ // 如果--index 就会陷入死循环
cur = cur->next;
}
return cur->val;
}
// 在链表最前面插入一个节点,插入完成后,新插入的节点为链表的新的头结点
void addAtHead(int val) {
LinkedNode* newNode = new LinkedNode(val);
newNode->next = _dummyHead->next;
_dummyHead->next = newNode;
_size++;
}
// 在链表最后面添加一个节点
void addAtTail(int val) {
LinkedNode* newNode = new LinkedNode(val);
LinkedNode* cur = _dummyHead;
while(cur->next != nullptr){
cur = cur->next;
}
cur->next = newNode;
_size++;
}
// 在第index个节点之前插入一个新节点,例如index为0,那么新插入的节点为链表的新头节点。
// 如果index 等于链表的长度,则说明是新插入的节点为链表的尾结点
// 如果index大于链表的长度,则返回空
// 如果index小于0,则置为0,作为链表的新头节点。
void addAtIndex(int index, int val) {
if (index > _size || index < 0) {
return;
}
LinkedNode* newNode = new LinkedNode(val);
LinkedNode* cur = _dummyHead;
while(index--) {
cur = cur->next;
}
newNode->next = cur->next;
cur->next = newNode;
_size++;
}
// 删除第index个节点,如果index 大于等于链表的长度,直接return,注意index是从0开始的
void deleteAtIndex(int index) {
if (index >= _size || index < 0) {
return;
}
LinkedNode* cur = _dummyHead;
while(index--) {
cur = cur ->next;
}
LinkedNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
_size--;
}
// 打印链表
void printLinkedList() {
LinkedNode* cur = _dummyHead;
while (cur->next != nullptr) {
cout << cur->next->val << " ";
cur = cur->next;
}
cout << endl;
}
private:
int _size;
LinkedNode* _dummyHead;
};
206 反转链表
题目链接:206. 反转链表 - 力扣(LeetCode)
很经典的一道单链表题目。要点是理解“单”链表的概念。单链表智能从头遍历到尾,如果哪个ListNode的链断了,或者被修改了,应该有变量保存相应的状态,不然单链表就没法往下继续操作了。我自己写的时候犯的错误就在于让temp=cur之后又分别操作了temp和cur,导致结果错误。
所以就想到了用两个指针,一个来遍历原链表,另一个表示新链表的节点在反转中链接的过程。
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;
}
};
递归的方法没完全看懂,今天结合视频再理解一下。
class Solution {
public:
ListNode* reverse(ListNode* pre,ListNode* cur){
if(cur == NULL) return pre;
ListNode* temp = cur->next;
cur->next = pre;
// 可以和双指针法的代码进行对比,如下递归的写法,其实就是做了这两步
// pre = cur;
// cur = temp;
return reverse(cur,temp);
}
ListNode* reverseList(ListNode* head) {
// 和双指针法初始化是一样的逻辑
// ListNode* cur = head;
// ListNode* pre = NULL;
return reverse(NULL, head);
}
};