你可以选择使用单链表或者双链表,设计并实现自己的链表。
单链表中的节点应该具备两个属性: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
的节点。
#include <iostream>
using namespace std;
// 定义链表类 MyLinkedList
class MyLinkedList {
public:
// 定义链表节点结构体
struct LinkedNode {
int val; // 节点值
LinkedNode* next; // 指向下一个节点的指针
// 构造函数,初始化节点的值和指针
LinkedNode(int val):val(val), next(nullptr){} //默认构造的节点指向nullptr
};
// 初始化链表
MyLinkedList() {
_dummyHead = new LinkedNode(0); // 创建一个虚拟头结点,初始值为0
_size = 0; // 初始化链表大小为0
}
// 获取第 index 个节点的值,如果 index 非法则返回 -1
int get(int index) {
if (index > (_size - 1) || index < 0) { // 判断 index 是否有效
return -1; // 如果 index 非法则返回 -1
}
LinkedNode* cur = _dummyHead->next; // 从第一个实际节点开始遍历
while(index--){ // 遍历到第 index个节点
cur = cur->next;
}
return cur->val; // 返回第 index 个节点的值
}
// 在链表头部插入一个新节点
void addAtHead(int val) {
LinkedNode* newNode = new LinkedNode(val); // 创建一个新节点
newNode->next = _dummyHead->next; // 新节点的 next 指向当前头节点 ,必须要先处理新节点的next指针,不然会丢失原链表头节点的引用
_dummyHead->next = newNode; // 虚拟头节点的 next 指向新节点
_size++; // 链表大小加1
}
// 在链表尾部添加一个新节点
void addAtTail(int val) {
LinkedNode* newNode = new LinkedNode(val); // 创建一个新节点
LinkedNode* cur = _dummyHead; // 从虚拟头节点开始遍历
while(cur->next != nullptr){ // 遍历到链表末尾
cur = cur->next;
}
cur->next = newNode; // 当前末尾节点的 next 指向新节点
_size++; // 链表大小加1
}
// 在第 index 个节点之前插入一个新节点
void addAtIndex(int index, int val) {
if(index > _size) return; // 如果 index 大于链表长度,则不插入
if(index < 0) index = 0; // 如果 index 小于0,则插入到头部
LinkedNode* newNode = new LinkedNode(val); // 创建一个新节点
LinkedNode* cur = _dummyHead; // 从虚拟头节点开始遍历
while(index--) { // 遍历到第 index 个节点的前一个节点
cur = cur->next;
}
newNode->next = cur->next; // 新节点的 next 指向当前节点的 next
cur->next = newNode; // 当前节点的 next 指向新节点
_size++; // 链表大小加1
}
// 删除第 index 个节点
void deleteAtIndex(int index) {
if (index >= _size || index < 0) { // 判断 index 是否有效
return; // 如果 index 非法则返回
}
LinkedNode* cur = _dummyHead; // 从虚拟头节点开始遍历
while(index--) { // 遍历到第 index 个节点的前一个节点
cur = cur->next;
}
LinkedNode* tmp = cur->next; // 保存要删除的节点
cur->next = cur->next->next; // 当前节点的 next 指向要删除节点的 next
delete tmp; // 删除节点
// 防止野指针
tmp=nullptr;
_size--; // 链表大小减1
}
// 打印链表
void printLinkedList() {
LinkedNode* cur = _dummyHead; // 从虚拟头节点开始遍历
while (cur->next != nullptr) { // 遍历整个链表
cout << cur->next->val << " "; // 打印当前节点的值
cur = cur->next; // 移动到下一个节点
}
cout << endl; // 打印换行
}
private:
int _size; // 链表大小
LinkedNode* _dummyHead; // 虚拟头节点
};
// 测试代码
int main() {
MyLinkedList linkedList;
linkedList.addAtHead(1); // 链表为1
linkedList.addAtTail(3); // 链表为1->3
linkedList.addAtIndex(1, 2); // 链表为1->2->3
cout << linkedList.get(1) << endl; // 返回2
linkedList.deleteAtIndex(1); // 链表为1->3
cout << linkedList.get(1) << endl; // 返回3
return 0;
}
https://chatgpt.com/share/6c324308-111c-4d61-9fa1-fe4686e1a65d