在C++中插入链表

#include <iostream>

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

// 在链表中插入新节点
void insertNode(Node** head, int value, int position) {
    // 创建新节点
    Node* newNode = new Node();
    newNode->data = value;
    newNode->next = nullptr;

    // 如果要插入的位置是链表的头部
    if (position == 0) {
        newNode->next = *head;
        *head = newNode;
        return;
    }

    // 找到插入位置的前一个节点
    Node* prevNode = *head;
    for (int i = 0; i < position - 1 && prevNode != nullptr; i++) {
        prevNode = prevNode->next;
    }

    // 如果找到了插入位置的前一个节点
    if (prevNode != nullptr) {
        newNode->next = prevNode->next;
        prevNode->next = newNode;
    } else {
        std::cout << "Invalid position" << std::endl;
    }
}

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

int main() {
    Node* head = nullptr;

    // 插入节点到链表
    insertNode(&head, 1, 0);  // 在头部插入节点
    insertNode(&head, 3, 1);  // 在位置1插入节点
    insertNode(&head, 2, 1);  // 在位置1插入节点

    // 打印链表
    printList(head);

    return 0;
}