203 移除链表元素
给你一个链表的头节点 head
和一个整数 val
,请你删除链表中所有满足 Node.val == val
的节点,并返回 新的头节点 。
输入:head = [1,2,6,3,4,5,6], val = 6
输出:[1,2,3,4,5]
方法一 构建新链表
创建新链表,将满足要求的插入到新链表当中
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
ListNode *dummy = new ListNode();
ListNode *tail = dummy;
ListNode *p = head;
while(p)
{
auto tmp = p->next;
if(p->val != val)
{
//尾插法
tail->next = p;
tail = tail->next;
}
else delete p;
p = tmp;
}
tail->next = nullptr;
return dummy->next;
}
};
方法二 在原链表上删除
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
的节点。
思路:
设计假头和尾节点,以及设计getPreNode函数,注意有些地方对tail的操作容易忽略,因此在每个函数中,都可以想一下头节点和尾节点是否需要改变,避免遗漏。如果确实容易遗漏对尾节点的操作,为避免产生错误,可去除尾节点,从头进行遍历来操作。
class MyLinkedList {
struct ListNode
{
int val = 0;
ListNode *next = nullptr;
ListNode() {}
ListNode(int x) { val = x; }
};
ListNode* dummy;
ListNode* tail;
int length;
public:
MyLinkedList() {
dummy = new ListNode(0);
tail = dummy;
length = 0;
}
ListNode* getPreNode(int index)
{
ListNode* back = dummy;
ListNode* front = dummy->next;
for(int i = 0; i < index && front != nullptr; i++)
{
back = front;
front = front->next;
}
return back;
}
int get(int index) {
if(index < 0 || index >= length) return -1;
return getPreNode(index)->next->val;
}
void addAtHead(int val) {
ListNode *p = new ListNode(val);
p->next = dummy->next;
dummy->next = p;
if(tail == dummy) //不要忽略
tail = p;
length++;
}
void addAtTail(int val) {
ListNode *p = new ListNode(val);
tail->next = p;
tail = p;
length++;
}
void addAtIndex(int index, int val) {
if(index > length) return;
else if(index == length)
{
addAtTail(val);
}
else if(index == 0)
{
addAtHead(val);
}
else
{
//cout << "val:" << val << endl;
ListNode* pre = getPreNode(index);
ListNode* p = new ListNode(val);
p->next = pre->next;
pre->next = p;
length++;
}
}
void deleteAtIndex(int index) {
if(index < 0 || index >= length) return;
ListNode* p = getPreNode(index);
if(p->next == tail) //不要忽略
tail = p;
ListNode* temp = p->next;
p->next = p->next->next;
length--;
delete temp;
}
};
206 反转链表
给你单链表的头节点 head
,请你反转链表,并返回反转后的链表。
思路:
创建新链表,使用头插法,即可完成链表反转
class Solution {
public:
ListNode* reverseList(ListNode* head) {
ListNode* dummy = new ListNode();
ListNode* tail = dummy;
ListNode* p = head;
//头插法
while(p)
{
auto tmp = p->next;
p->next = dummy->next;
dummy->next = p;
p = tmp;
}
return dummy->next;
}
};
注意:本题虽然创建了新链表,但仅创建了假头,其余则是从原链表中取出,放置到新链表中,因此空间复杂度为O(1)
其余方法包括双指针、递归,可参考代码随想录:
代码随想录 (programmercarl.com)