数据结构链表

Question

我已经实现了一个自定义链接列表，但我找不到我预期的结果，因为它无法纠正问题。

package LinkedList;
public class Node {
protected Node nextNode;
protected int data;
public Node(int data) {
this.data = data;
}
}

package LinkedList;
public class LinkedList {
private Node head;
public void insertLast(int data) {
Node new_node = new Node(data);
if(head == null) {
head = new_node;
}else {
Node n = head;
while(n.nextNode != null) {
n = n.nextNode;
}   
n.nextNode = new_node;
}   
}
public void insertFirst(int data) {
Node new_node = new Node(data);
if(head == null) {
head = new_node;
}else {
Node node = head;
head = new_node;
head.nextNode = node;
}
}
public void insertAt(int index,int data) {
int x = 1;
Node node = head;
while(node != null) {
if(index == 0) {
insertFirst(data);
break;
}else {
node = node.nextNode;
if( index == x ) {
Node new_node = new Node(data);
Node current = node;
node = new_node;
node.nextNode = current;
break;
}
x++;
}
}
}
public void print() {
Node node = head;
while(node != null) {
System.out.println(node.data);
node = node.nextNode;
}
}

}

package Test;
import LinkedList.*;
public class Main {
public static void main(String[] args) {
LinkedList list = new LinkedList();
list.insertLast(100);
list.insertLast(200);
list.insertLast(300);
list.insertLast(400);
list.insertLast(500);
list.insertAt(1, 50);
list.print();
}
}

将元素添加到特定位置时会出现此问题。 打印结果时，该元素不会添加到列表中，我很难调试代码以解决问题。

Answer 1

您在insertAt()中的插入完全错过了新节点。您希望在两个节点之间添加元素，以便从node1 -> node2更改为node1 -> new_node -> node2。您需要将节点 1 的下一个节点修改为新节点，并确保新节点指向节点 2。尝试这样的代码：

public void insertAt(int index,int data) {
//assumes index <= size of linked list
int curr = 1;
if(index < 0)
return;
if(index == 0){
insertFirst(data);
return;
}
Node currNode = head;
while( curr < index){
currNode = currNode.nextNode;
curr++;
}
Node newNode = new Node(data);
newNode.nextNode = currNode.nextNode;
currNode.nextNode = newNode;
}