java - 在链表中的特定位置插入节点

Question

我得到了指向链表头节点的指针、要添加到列表中的整数以及必须插入整数的位置。在所需位置插入此节点后，我需要返回头节点。

我编写的代码由于某种原因无法正常工作并进入无限循环。

  class Node {
     int data;
     Node next;
  }


Node InsertNth(Node head, int data, int position) {
    int count = 0;
    Node node = head;
    Node prev = null;
    while(count != position){
      count++;
      node = node.next;
      prev = node;
    }

    Node newNode = new Node();
    newNode.data = data;


    newNode.next = node;
    if(count == 0){
          head = newNode;
       }else{
          prev.next = newNode;
    }

    return head;          
}

Answer 1

node = node.next;
prev = node;

这应该是相反的顺序

prev = node;
node = node.next;

而且很多情况下代码不会检查，例如指定的位置是否大于链表的大小。尝试重写代码，您还可以提及您使用哪些值来测试功能。我认为 prev=node 应该抛出一个异常，因为 prev 没有被初始化