C++链表插入实现

#include <iostream>

// 定义链表节点的结构
struct Node {
    int data; // 节点的数据
    Node* next; // 指向下一个节点的指针
};

// 在链表的指定位置插入新节点
void insertNode(Node** head, int position, int newData) {
    // 创建新节点
    Node* newNode = new Node();
    newNode->data = newData;
    newNode->next = nullptr;
    
    // 如果插入位置为0，即在头部插入节点
    if (position == 0) {
        newNode->next = *head;
        *head = newNode;
        return;
    }
    
    // 找到插入位置的前一个节点
    Node* prevNode = *head;
    for (int i = 0; prevNode != nullptr && i < position - 1; i++) {
        prevNode = prevNode->next;
    }
    
    // 如果插入位置超出链表长度，则直接在末尾插入节点
    if (prevNode == nullptr) {
        prevNode->next = newNode;
        return;
    }
    
    // 将新节点的指针指向插入位置的后一个节点
    newNode->next = prevNode->next;
    // 将插入位置的前一个节点的指针指向新节点
    prevNode->next = newNode;
}

// 打印链表
void printList(Node* node) {
    while (node != nullptr) {
        std::cout << node->data << " ";
        node = node->next;
    }
    std::cout << std::endl;
}

int main() {
    // 创建一个链表
    Node* head = nullptr;
    insertNode(&head, 0, 1); // 在头部插入节点
    insertNode(&head, 1, 3); // 在第二个位置插入节点
    insertNode(&head, 1, 2); // 在第二个位置插入节点
    
    // 打印链表
    printList(head); // 输出：1 2 3
    
    return 0;
}