/**
 * 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;
}