给你一个链表的头节点 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 输出:[]
提示:
- 列表中的节点数目在范围
[0, 104]
内 1 <= Node.val <= 50
0 <= val <= 50
#include "iostream"
#include <vector>
#include <algorithm>
using namespace std;
struct ListNode {
int val;
ListNode* next;
ListNode(int val) :val(val), next(nullptr) {};
};
ListNode* head=nullptr;
void insertAtHead(int x) {
ListNode* newNode = new ListNode(x);
newNode->next = head;
head = newNode;
}
void insertAtTail(int x) {
ListNode* newNode = new ListNode(x);
ListNode* p = head;
if (p == nullptr) {
head = newNode;
}
else {
while (p->next != nullptr) p = p->next;
p->next = newNode;
}
}
void printList() {
ListNode* p = head;
if (p == nullptr) {
cout << "Empty List" << endl; // 如果链表为空,输出提示信息
return;
}
while (p != nullptr) {
cout << p->val << " "; // 添加空格作为分隔符
p = p->next;
}
cout << endl; // 添加换行
}
class Solution {
public:
ListNode* removeElements(ListNode* head, int val) {
// 删除头结点
while (head != NULL && head->val == val) { // 注意这里不是if
ListNode* tmp = head;
head = head->next;
delete tmp;
}
// 删除非头结点
ListNode* cur = head;
while (cur != NULL && cur->next != NULL) {
if (cur->next->val == val) {
ListNode* tmp = cur->next;
cur->next = cur->next->next;
delete tmp;
}
else {
cur = cur->next;
}
}
return head;
}
};
int main() {
int n;
cin >> n;
for (int i = 0; i < n; i++) {
int a;
cin >> a;
insertAtTail(a);
}
Solution p;
p.removeElements(head, 2);
printList();
}
你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
提示:
0 <= index, val <= 1000
- 请不要使用内置的 LinkedList 库。
- 调用
get
、addAtHead
、addAtTail
、addAtIndex
和deleteAtIndex
的次数不超过2000
。
class MyLinkedList {
public:
// 定义链表ListNode
class ListNode {
public:
int val;
ListNode* next;
ListNode(int x) :val(x), next(nullptr) {};
};
private:
ListNode* head; // 指向链表的第一个节点
int size; // 链表长度
public:
// 默认构造函数,初始化链表
MyLinkedList() : head(nullptr), size(0) {}
int get(int index) {
if (index < 0 || index >= size) return -1; // 返回 -1 表示无效索引
ListNode* p = head;
for (int i = 0; i < index; i++) {
p = p->next;
}
return p->val;
}
void addAtHead(int val) { //头接法,在头节点前面增加新节点,并更新头节点位置
ListNode* newNode = new ListNode(val);
newNode->next = head;
head = newNode;
size++;
}
void addAtTail(int val) { //尾接法,找到链表最后一个节点,在其后面新添节点
ListNode* newNode = new ListNode(val);
if (head == nullptr) {
head = newNode;
size++;
return;
}
ListNode* p = head;
while (p->next != nullptr) {
p = p->next;
}
p->next = newNode;
size++;
}
void addAtIndex(int index, int val) { //在链表特定位置新添一个节点
if (index <= 0) {
addAtHead(val);
return;
}
if (index > size) {
return;
}
if (index == size) {
addAtTail(val);
return;
}
ListNode* p = head;
for (int i = 0; i < index - 1; i++) {
p = p->next;
}
ListNode* newNode = new ListNode(val);
newNode->next = p->next;
p->next = newNode;
size++;
}
void deleteAtIndex(int index) { //在链表特定位置删除节点
if (index < 0 || index >= size) return; // 无效索引,直接返回
if (index == 0) {
ListNode* p = head;
head = head->next;
delete p;
size--;
return;
}
ListNode* p = head;
for (int i = 0; i < index - 1; i++) {
p = p->next;
}
ListNode* q = p->next;
p->next = q->next;
delete q;
size--;
}
void printList() { //打印链表
ListNode* p = head;
if (p == nullptr) {
cout << "List is empty" << endl;
return;
}
while (p != nullptr) {
cout << p->val << " ";
p = p->next;
}
cout << endl;
}
};
给你单链表的头节点 head
,请你反转链表,并返回反转后的链表。
示例 1:
输入:head = [1,2,3,4,5] 输出:[5,4,3,2,1]
示例 2:
输入:head = [1,2] 输出:[2,1]
示例 3:
输入:head = [] 输出:[]
提示:
- 链表中节点的数目范围是
[0, 5000]
-5000 <= Node.val <= 5000
进阶:链表可以选用迭代或递归方式完成反转。你能否用两种方法解决这道题?
class Solution {
public:
class ListNode {
public:
int val;
ListNode* next;
ListNode(int x) :val(x), next(nullptr) {};
};
ListNode* reverseList(ListNode* head) {
if (head==NULL||head->next == NULL) return head; //如果链表为空或者只有一个节点,就返回原链表
ListNode* cur=head; //当前节点
ListNode* pre=nullptr; //当前节点的前一个节点
ListNode* nxt=nullptr; //当前节点后一个节点
while (cur != NULL) {
nxt = cur->next; //更新nxt节点为当且节点后一个节点
cur->next = pre; //当前节点指向前一个节点
pre = cur; //更新pre节点
cur = nxt; //更新当前节点
}
return pre; //最后的pre节点就是反转链表后的头节点
}
};