题目:在O(1)时间内删除链表节点。
方法:把需要删除的这个节点的下一个节点的值复制给当前节点,然后把需要删除的当前节点的下一个节点指向下下节点。
Java代码如下:
public static void main(String[] args) {
ListNode head =new ListNode();
head.setValue(0);
ListNode next1 =new ListNode();
next1.setValue(1);
ListNode next2 =new ListNode();
next2.setValue(2);
ListNode next3 =new ListNode();
next3.setValue(3);
ListNode next4 =new ListNode();
next4.setValue(4);
head.setNext(next1);
next1.setNext(next2);
next2.setNext(next3);
next3.setNext(next4);
//不是头节点也不是尾节点 验证删除next1,head的next是不是val为2。
ListNode result = delete(head,next1);
System.out.println(result.getNext().getValue());
//头节点 验证删除头节点是不是直接返回null了。
ListNode result1 = delete(head,head);
System.out.println(result1);
//尾节点 验证next3的下一个节点是不是为null。
ListNode result2 = delete(head,next4);
System.out.println(result2.getNext().getNext().getNext().getNext());
}
public static ListNode delete(ListNode head,ListNode val){
if (head == val){
return null;
}
ListNode next = head.getNext();
if (val.getNext() == null){
while (next != null){
if (next == val){
next.setNext(null);
return head;
}
next = next.getNext();
}
}
ListNode vNext = val.getNext();
val.setValue(vNext.getValue());
val.setNext(vNext.getNext());
return head;
}