在链表中插入节点时发生逻辑错误（Java）

Question

我试图学习在链表中插入一个节点（并返回头部），但由于某种原因，它是不正确的。

这是我的方法：

1.使用所需数据创建新节点

2.如果我们想在开头插入它，将这个新节点指向头部并返回新节点

3.否则，循环到要插入节点的位置

-到达后，将要插入的节点指向当前节点的下一个节点

-将当前节点指向要插入的节点

-把脑袋还给我

为什么这不起作用？非常感谢！

Node InsertNth(Node head, int data, int position) {
    Node node = new Node();
    node.data = data;

    if (position == 0) {
        node.next = head;
        return node;
    }
    else {
        Node curr = head;
        int currPos = 0;

        while (currPos < position) {
            curr = curr.next;
            currPos++;
        }
        node.next = curr.next;
        curr.next = node;
    }
    return head;
}

Answer 1

如果将节点插入头部，则需要将头部设置为要插入的节点。

Node InsertNth(Node head, int data, int position) {
Node node = new Node();
node.data = data;

if (position == 0) {
    node.next = head;
    head = node;
}
else {
    Node curr = head;
    int currPos = 0;

    while (currPos < position) {
        curr = curr.next;
        currPos++;
    }
    node.next = curr.next;
    curr.next = node;
}
return head;

}

Answer 2

如果在curr节点之后插入新节点，则此循环会使一个节点走得太远。

while (currPos < position) {
     curr = curr.next;
     currPos++;
}

您可以通过用笔和纸或调试器遍历代码轻松解决这个问题。