97 lines
2.1 KiB
C++
97 lines
2.1 KiB
C++
/**
|
|
* File: linked_list.cpp
|
|
* Created Time: 2022-11-25
|
|
* Author: krahets (krahets@163.com)
|
|
*/
|
|
|
|
#include "../utils/common.hpp"
|
|
|
|
/* 在链表的节点 n0 之后插入节点 P */
|
|
void insert(ListNode *n0, ListNode *P)
|
|
{
|
|
ListNode *n1 = n0->next; // n0 -> n1
|
|
P->next = n1; // P -> n1
|
|
n0->next = P; // n0 -> P
|
|
}
|
|
|
|
/* 删除链表的节点 n0 之后的首个节点 */
|
|
void remove(ListNode *n0)
|
|
{
|
|
if (n0->next == nullptr)
|
|
return; // n0 之后没有节点
|
|
// n0 -> P -> n1
|
|
ListNode *P = n0->next;
|
|
ListNode *n1 = P->next;
|
|
n0->next = n1;
|
|
// 释放内存
|
|
delete P;
|
|
}
|
|
|
|
/* 访问链表中索引为 index 的节点 */
|
|
ListNode *access(ListNode *head, int index)
|
|
{
|
|
for (int i = 0; i < index; i++)
|
|
{
|
|
if (head == nullptr)
|
|
return nullptr;
|
|
head = head->next;
|
|
}
|
|
return head;
|
|
}
|
|
|
|
/* 在链表中查找值为 target 的首个节点 */
|
|
int find(ListNode *head, int target)
|
|
{
|
|
int index = 0;
|
|
while (head != nullptr)
|
|
{
|
|
if (head->val == target) // 找到节点
|
|
return index; // 返回索引
|
|
head = head->next; // 更新节点
|
|
index++; // 更新索引
|
|
}
|
|
return -1; // 未找到节点, 返回 -1
|
|
}
|
|
|
|
/* Driver Code */
|
|
int main()
|
|
{
|
|
/* 初始化链表 */
|
|
// 初始化各个节点
|
|
ListNode *n0 = new ListNode(1); // 创建一个节点, 值为1
|
|
ListNode *n1 = new ListNode(3);
|
|
ListNode *n2 = new ListNode(2);
|
|
ListNode *n3 = new ListNode(5); // 创建一个节点, 值为5
|
|
ListNode *n4 = new ListNode(4);
|
|
// 构建节点之间的引用
|
|
n0->next = n1;
|
|
n1->next = n2;
|
|
n2->next = n3;
|
|
n3->next = n4;
|
|
cout << "初始化的链表为" << endl;
|
|
printLinkedList(n0);
|
|
|
|
/* 插入节点 */
|
|
insert(n0, new ListNode(0));
|
|
cout << "插入节点后的链表为" << endl;
|
|
printLinkedList(n0);
|
|
|
|
/* 删除节点 */
|
|
remove(n0);
|
|
cout << "删除节点后的链表为" << endl;
|
|
printLinkedList(n0);
|
|
|
|
/* 访问节点 */
|
|
ListNode *node = access(n0, 3);
|
|
cout << "链表中索引 3 处的节点的值 = " << node->val << endl;
|
|
|
|
/* 查找节点 */
|
|
int index = find(n0, 2);
|
|
cout << "链表中值为 2 的节点的索引 = " << index << endl;
|
|
|
|
// 释放内存
|
|
freeMemoryLinkedList(n0);
|
|
|
|
return 0;
|
|
}
|