我一直在研究关于黑客排名的Linkedlist问题,目前我正在解决一个问题,这个问题要求你在一个排序好的双链表中插入一个节点。
这是我用Java写的逻辑
Node SortedInsert(Node head,int data) {
Node newNode = new Node();
Node temp = head;
newNode.data = data;
newNode.next = null;
newNode.prev = null;
if(head == null){
head = newNode;
}else{
while(data > temp.data && temp.next != null){
temp = temp.next;
}
if(temp == head){
newNode.next = temp;
temp.prev = newNode;
head = newNode;
}else if(data > temp.data){
newNode.prev = temp;
temp.next = newNode;
}else{
newNode.prev = temp.prev;
newNode.next = temp;
temp.prev.next = newNode;
temp.prev = newNode;
}
}
return head;
}
这是我得到的错误。
Your Output (stdout)
Wrong Answer!
Some possible errors:
1. You returned a NULL value from the function.
2. There is a problem with your logic
Wrong Answer!
Some possible errors:
1. You returned a NULL value from the function.
2. There is a problem with your logic
我不知道我做错了什么。我真的很想知道我哪里出错了。我知道在网上很容易找到答案,但我想如果有人能纠正我的错误,我会学得最好。
问题是您总是将第二个元素插入到列表中的第一个元素之前。考虑下面的例子:
让链表初始为空。现在按照算法插入1
。触发head == null
,现在head
指向newNode
。
x<-1->x
|
HEAD
现在,您尝试在列表中插入2
。您将看到,while
循环结束,temp
现在指向head
,触发后面的if条件(if(temp == head)
)。
x<-1->x
|
HEAD, temp
在 temp
之前插入2
(不正确!)。
x<-2<=>1->x
|
HEAD
交换条件的顺序应该可以解决这个问题:
if(data > temp.data) { // First, check if you need to insert at the end.
newNode.prev = temp;
temp.next = newNode;
} else if(temp == head) { // Then, check if you need to insert before head.
newNode.next = temp;
temp.prev = newNode;
head = newNode;
} else { // Otherwise, insert somewhere in the middle.
newNode.prev = temp.prev;
newNode.next = temp;
temp.prev.next = newNode;
temp.prev = newNode;
}