203. 移除链表元素
方法一:直接移除
思路
直接移除满足条件的节点,但是对于头结点和非头结点的移除逻辑不同,所以要分别进行操作。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( 1 ) O(1) O(1)
C++代码
/**
* 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) {
// 删除头结点
while (head != nullptr && head->val == val) {
ListNode* tmp = head;
head = head->next;
delete tmp;
}
// 删除非头结点
ListNode* cur = head;
while (cur != nullptr && cur->next != nullptr) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
return head;
}
};
方法二:使用虚拟头结点
思路
方法一需要分别处理头结点和非头结点,可以首先在链表头设置一个虚拟头结点,这样就能统一原本的头结点和非头结点的操作,使代码逻辑更加简洁,最后再返回虚拟头结点的next。
- 时间复杂度: O ( n ) O(n) O(n)
- 空间复杂度: O ( 1 ) O(1) O(1)
C++代码
/**
* 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* preHead = new ListNode(0, head);
ListNode* cur = preHead;
while (cur->next != nullptr) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
} else {
cur = cur->next;
}
}
head = preHead->next;
delete preHead;
return head;
}
};
看完讲解的思考
无
代码实现遇到的问题
太久没处理各种数据结构了,对于指针的操作一开始没理清楚。
707. 设计链表
方法一:单链表
思路
经典单链表的操作,重点在于设置虚拟头结点,不设置虚拟头结点的话代码逻辑会很复杂。
- 时间复杂度:get是 O ( i n d e x ) O(index) O(index),addAtHead是 O ( 1 ) O(1) O(1),addAtTail是 O ( n ) O(n) O(n),addAtIndex是 O ( i n d e x ) O(index) O(index),deleteAtIndex是 O ( i n d e x ) O(index) O(index)。
- 空间复杂度:不考虑链表本身的存储空间的话,所有操作都为 O ( 1 ) O(1) O(1)。
C++代码
class MyLinkedList {
public:
// 定义链表节点结构体
struct LinkedNode {
int val;
LinkedNode* next;
LinkedNode(int val) : val(val), next(nullptr) {}
};
MyLinkedList() {
this->_size = 0;
this->_dummyHead = new LinkedNode(-1);
}
int get(int index) {
if (index < 0 || index >= this->_size) return -1;
LinkedNode* cur = this->_dummyHead->next;
for (int i = 0; i < index; ++i) cur = cur->next;
return cur->val;
}
void addAtHead(int val) {
LinkedNode* newNode = new LinkedNode(val);
newNode->next = this->_dummyHead->next;
this->_dummyHead->next = newNode;
++this->_size;
}
void addAtTail(int val) {
LinkedNode* newNode = new LinkedNode(val);
LinkedNode* cur = this->_dummyHead;
while (cur->next != nullptr) cur = cur->next;
cur->next = newNode;
++this->_size;
}
void addAtIndex(int index, int val) {
if (index >= 0 && index <= this->_size) {
LinkedNode* cur = this->_dummyHead;
LinkedNode* newNode = new LinkedNode(val);
for (int i = 0; i < index; ++i) cur = cur->next;
newNode->next = cur->next;
cur->next = newNode;
++this->_size;
}
}
void deleteAtIndex(int index) {
if (index >= 0 && index < this->_size) {
LinkedNode* cur = this->_dummyHead;
for (int i = 0; i < index; ++i) cur = cur->next;
LinkedNode* delNode = cur->next;
cur->next = delNode->next;
delete delNode;
delNode = nullptr;
--this->_size;
}
}
private:
int _size;
LinkedNode* _dummyHead;
};
/**
* Your MyLinkedList object will be instantiated and called as such:
* MyLinkedList* obj = new MyLinkedList();
* int param_1 = obj->get(index);
* obj->addAtHead(val);
* obj->addAtTail(val);
* obj->addAtIndex(index,val);
* obj->deleteAtIndex(index);
*/
方法二:双向链表
思路
经典双向链表的操作,重点在于设置虚拟头结点和虚拟尾结点,不设置虚拟头尾结点的话代码逻辑会很复杂。
- 时间复杂度:get是 O ( m i n { i n d e x , s i z e − 1 − i n d e x } ) O(min\{index, size-1-index\}) O(min{index,size−1−index}),addAtHead是 O ( 1 ) O(1) O(1),addAtTail是 O ( 1 ) O(1) O(1),addAtIndex是 O ( m i n { i n d e x , s i z e − i n d e x } ) O(min\{index, size-index\}) O(min{index,size−index}),deleteAtIndex是 O ( m i n { i n d e x , s i z e − i n d e x } ) O(min\{index, size-index\}) O(min{index,size−index})。
- 空间复杂度:不考虑链表本身的存储空间的话,所有操作都为
O
(
1
)
O(1)
O(1)。但是双向链表相对于单链表需要额外的存储空间来存储
prev
。
C++代码
class MyLinkedList {
public:
// 定义双链表结点结构体
struct DLinkedNode {
int val;
DLinkedNode* next;
DLinkedNode* prev;
DLinkedNode(int val) : val(val), next(nullptr), prev(nullptr) {}
DLinkedNode(int val, DLinkedNode* next, DLinkedNode* prev) : val(val), next(next), prev(prev) {}
};
MyLinkedList() {
this->_size = 0;
this->head = new DLinkedNode(-1);
this->tail = new DLinkedNode(-1);
this->head->next = this->tail;
this->tail->prev = this->head;
}
int get(int index) {
if (index < this->_size && index >= 0) {
DLinkedNode* cur;
if (index < (this->_size - 1) / 2) {
// 要获取的结点位于链表前半段
cur = this->head->next;
while (index--) cur = cur->next;
} else {
// 要获取的结点位于链表后半段
cur = this->tail->prev;
while (index++ < this->_size - 1) cur = cur->prev;
}
return cur->val;
}
return -1;
}
void addAtHead(int val) {
DLinkedNode* newNode = new DLinkedNode(val, this->head->next, this->head);
this->head->next->prev = newNode;
this->head->next = newNode;
++this->_size;
}
void addAtTail(int val) {
DLinkedNode* newNode = new DLinkedNode(val, this->tail, this->tail->prev);
this->tail->prev->next = newNode;
this->tail->prev = newNode;
++this->_size;
}
void addAtIndex(int index, int val) {
if (index <= this->_size && index >= 0) {
if (index <= this->_size / 2) {
// 要增加的结点的位置位于链表前半段
DLinkedNode* cur = this->head;
while (index--) cur = cur->next;
DLinkedNode* newNode = new DLinkedNode(val, cur->next, cur);
cur->next->prev = newNode;
cur->next = newNode;
} else {
// 要增加的结点的位置位于链表后半段
DLinkedNode* cur = this->tail;
while (index++ < this->_size) cur = cur->prev;
DLinkedNode* newNode = new DLinkedNode(val, cur, cur->prev);
cur->prev->next = newNode;
cur->prev = newNode;
}
++this->_size;
}
}
void deleteAtIndex(int index) {
if (index < this->_size && index >= 0) {
DLinkedNode* delNode;
if (index <= this->_size / 2) {
// 要删除的结点位于链表前半段
DLinkedNode* cur = this->head;
while (index--) cur = cur->next;
delNode = cur->next;
delNode->next->prev = cur;
cur->next = delNode->next;
} else {
// 要删除的结点位于链表后半段
DLinkedNode* cur = this->tail;
while (index++ < this->_size - 1) cur = cur->prev;
delNode = cur->prev;
delNode->prev->next = cur;
cur->prev = delNode->prev;
}
delete delNode;
delNode = nullptr;
--this->_size;
}
}
private:
int _size;
DLinkedNode* head;
DLinkedNode* tail;
};
/**
* Your MyLinkedList object will be instantiated and called as such:
* MyLinkedList* obj = new MyLinkedList();
* int param_1 = obj->get(index);
* obj->addAtHead(val);
* obj->addAtTail(val);
* obj->addAtIndex(index,val);
* obj->deleteAtIndex(index);
*/
看完讲解的思考
划重点:单链表要设置虚拟头结点!双向链表要设置虚拟头尾结点!
可以大大简化代码逻辑。
对某个结点进行删除或者在某个结点之后添加新结点的操作,一定是遍历到目标结点的前一个结点进行操作。例如:要删除第5个结点,则要遍历到第4个结点,然后对第5个结点进行删除;要在第5个位置添加结点,则要遍历到第4个结点,然后在后面添加新结点。
该题还可以用其他链表实现,例如循环链表,但目前水平有限,之后二刷再来填坑。
代码实现遇到的问题
因为没有设置虚拟头结点,导致代码逻辑混乱,一堆bug…
最后的碎碎念
今日的数据结构链表太久没写了,折腾了半天,第二题搞了一晚上,今天就刷两题吧。