代码随想录算法训练营第三天|链表理论基础、203. 移除链表元素、707. 设计链表、 206. 反转链表
代码随想录打卡的第三天,今天首先学习链表的基础知识
链表是有数据域和指针域组成的线性结构,具体如图所示:
head是链表的入口也称为头结点,链表的结尾是空指针null
链表的类型
根据链表指针的不同拓扑结构,还可以将链表分类为以下几种:
- 单链表
- 双链表
- 循环链表
链表的存储方式
与数组的连续存储不同,链表在内存中不是连续分布的,所以链表散乱分布在内存中。
链表的定义
单链表的定义方式如下:
链表的定义分为两步
- 定义数据域和指针域
- 定义构造函数(当然不写构造函数编译器也会自动生成,但是初始化的时候不太方便)
具体代码如下:
// 单链表
struct ListNode{
int val; // 数据域
ListNode *next; // 指针域
ListNode(int x) : val(x), next(NULL){} // 节点的构造函数
};
链表初始化
ListNode* head = new ListNode(5);
链表的操作
删除节点
如图所示,如果要删除D节点,需要让C的next指针指向E,并释放D节点的内存
todo:先后顺序
添加节点
如果所示,如果要添加节点F则需要将节点C的next指针指向F节点,并将F节点的next指针指向D节点。
性能分析
对比数组和链表的性能如图所示:
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
输出:[]
针对该题目,删除的时候有两种操作,第一种直接在原来的链表上进行删除,但是对于头节点需要单独写一段逻辑特殊对待;第二种方法是虚拟指针法,该方法可以统一所有节点的删除操作,无论是头结点还是一般节点,接下来分别介绍一下两种方式。
思路1
采用直接删除的方式进行,总体来说分为两步
- 判断头结点是否需要删除,如果删除,直接将指针指向头结点的下一个节点,并释放头结点
- 判断一般节点是否需要删除,如果删除,将待删除节点前一个节点的指针指向待删除节点的后一个节点,并释放待删除的节点
具体代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
// 删除头节点
// 判断头结点不为空
// 判断是否等于val
while(head!=NULL && head->val==val){
// 记录head节点
ListNode* temp = head;
head = head->next;
delete temp;
}
// 删除非头节点
ListNode* cur = head;
// while循环中的条件是决定是否继续移动指针的
// 如果当前节点为空 说明在删除头结点的循环中已经将链表中的全部节点删除掉了
// 如果当前节点的下一个节点为空 说明在删除头节点的循环后只剩下一个节点了
// 或者删除到最后 也是通过cur->next!=NULL来判断是否需要删除
while(cur!=NULL && cur->next!=NULL){
// 判断节点是否需要删除
if(cur->next->val == val){
// 存储待删除节点
ListNode* temp = cur->next;
// 将当前节点指向删除节点的后一个节点
cur->next = cur->next->next;
// 释放当前节点
delete temp;
}else{
cur = cur->next;
}
}
return head;
}
};
要注意在删除一般节点的时候,不仅要判断当前节点是否为空,还要判断当前节点的下一个节点是否为空,因为一旦下一个节点为空,也就没有了要移除的元素。这一点在自己写代码的时候没有考虑到。
- 时间复杂度: O(n)
- 空间复杂度: O(1)
思路2
还可以采用虚拟头结点的方式来写,这样写的好处是可以统一逻辑处理所有的节点了
具体示意图如下
采用虚拟头节点的方式删除节点的具体步骤如下:
- 添加虚拟头节点
- 判断是否需要删除节点,若需要,则按照一般步骤进行删除
- 返回虚拟头结点next
- 释放虚拟头节点
因为虚拟头节点的加入,使得头结点也变成了一般类型的节点了
具体代码如下:
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
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* temp = cur->next;
cur->next = cur->next->next;
delete temp;
}else{
cur=cur->next;
}
}
head = DummyHead->next;
delete DummyHead;
return head;
}
};
要注意的是在创建虚拟头结点的时候要使用new关键字ListNode* DummyHead = new ListNode(0)
- 时间复杂度: O(n)
- 空间复杂度: O(1)
707. 设计链表
题目
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
思路
本题直接采用虚拟头节点的方式统一处理
注意,可以在类中维护两个成员
- int _size 链表中节点的个数
- LinkedNode* _dummyHead 虚拟头节点
这样在后续编程中会更加方便
class MyLinkedList {
public:
// 定义链表节点结构体
struct LinkedNode{
int val;
LinkedNode* next;
LinkedNode(int x) : val(x) , next(NULL){}
};
// 初始化链表
MyLinkedList() {
// 初始化虚拟头节点
_dummyHead = new LinkedNode(0);
// 初始化size
_size = 0;
printLinkedList();
}
// 获取下标为index的节点的值,如果下标无效,返回-1
// 所有下标从0开始,如果索引为index实际上是第index-1个节点
int get(int index) {
// 判断输入是否合法
if(index > (_size-1) || index < 0){
return -1;
}
LinkedNode* cur = _dummyHead;
// 获取index位置的节点值
for(int i=0;i<index;i++){
cur = cur->next;
}
// printLinkedList();
return cur->next->val;
}
void addAtHead(int val) {
LinkedNode* newNode = new LinkedNode(val);
newNode->next = _dummyHead->next;
_dummyHead->next = newNode;
_size++;
// printLinkedList();
}
void addAtTail(int val) {
LinkedNode* lastNode = new LinkedNode(val);
LinkedNode* cur = _dummyHead;
for(int i=0;i<_size;i++){
cur = cur -> next;
}
cur->next = lastNode;
_size++;
// printLinkedList();
}
// 在第index个节点之前插入一个新节点,例如index为0,那么新插入的节点为链表的新头节点。
// 如果index 等于链表的长度,则说明是新插入的节点为链表的尾结点
// 如果index大于链表的长度,则返回空
// 如果index小于0,则在头部插入节点
void addAtIndex(int index, int val) {
if(index == _size)
{
addAtTail(val);
}else if(index > (_size-1)){
return;
}else{
LinkedNode* cur = _dummyHead;
for(int i=0;i<index;i++){
cur = cur->next;
}
LinkedNode* newNode = new LinkedNode(val);
newNode->next = cur->next;
cur->next = newNode;
_size++;
}
// printLinkedList();
}
void deleteAtIndex(int index) {
if(index >=0 && index <=(_size-1)){
LinkedNode* cur = _dummyHead;
for(int i=0;i<index;i++){
cur = cur->next;
}
LinkedNode* temp = cur->next;
cur->next = cur->next->next;
delete temp;
_size--;
}
// printLinkedList();
}
// 打印链表
void printLinkedList() {
LinkedNode* cur = _dummyHead;
while (cur->next != nullptr) {
cout << cur->next->val << " ";
cur = cur->next;
}
cout << endl;
}
private:
int _size;
LinkedNode* _dummyHead;
};
注意for循环和while循环的次数
for(int i = 0; i < num ; i++)
{
// 循环num次
}
while(num--)
{
// 循环num次
}
206. 反转链表
题目
给你单链表的头节点head
,请你反转链表,并返回反转后的链表。
示例 1:
输入:head = [1,2,3,4,5]
输出:[5,4,3,2,1]
示例 2:
输入:head = [1,2]
输出:[2,1]
示例 3:
输入:head = []
输出:[]
思路
这里发现可以在链表的开头和结尾都加上null,这样可以更容易地理解。
在代码里面维护三个变量,分别用于遍历链表cur、存储cur下一个节点temp、存储cur上一个节点pre
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode* cur = head;
ListNode* pre = NULL;
ListNode* temp = NULL;
while(cur){
temp = cur->next;
cur->next =pre;
pre = cur;
cur = temp;
}
return pre;
}
};
这道题只理解了双指针的写法,其他两种方法后续再研究。
- 时间复杂度: O(n)
- 空间复杂度: O(1)
🤗 总结归纳
该文档详细介绍了链表的理论基础,包括链表的存储方式、定义、操作、性能分析等。同时,通过三个具体的编程题目(203.移除链表元素、707.设计链表、206.反转链表)进行了实践,解释了如何在链表中添加、删除和反转节点。