数据结构笔记——单链表的插入

代码实现

#include <iostream>

// 定义链表节点结构
struct Node {
    int data;
    Node* next;
};

// 头插法
void insertAtBeginning(Node*& head, int newData) {
    Node* newNode = new Node;
    newNode->data = newData;
    newNode->next = head;
    head = newNode;
}

// 在链表指定位置后插入节点
void insertAfterNode(Node* prevNode, int newData) {
    if (prevNode == nullptr) {
        std::cout << "Previous node cannot be null." << std::endl;
        return;
    }
    Node* newNode = new Node;
    newNode->data = newData;
    newNode->next = prevNode->next;
    prevNode->next = newNode;
}

// 输出链表元素
void displayList(Node* head) {
    Node* currentNode = head;
    while (currentNode != nullptr) {
        std::cout << currentNode->data << " ";
        currentNode = currentNode->next;
    }
    std::cout << std::endl;
}

int main() {
    Node* head = nullptr;

    // 在链表头部插入节点
    insertAtBeginning(head, 3);
    insertAtBeginning(head, 2);
    insertAtBeginning(head, 1);

    // 输出链表元素
    std::cout << "链表内容：";
    displayList(head);

    // 在第二个节点后插入新节点
    Node* secondNode = head->next;
    insertAfterNode(secondNode, 4);

    // 输出链表元素
    std::cout << "插入节点后的链表内容：";
    displayList(head);

    return 0;
}